Refactor agent version handling and improve logging

- Updated the `update_agent` function to access `version_number` and `version_name` directly from the `new_version` object, enhancing clarity and consistency in version management.
- Improved logging statements to reflect the changes in how version details are accessed, ensuring accurate log messages for created versions.
This commit is contained in:
mykonos-ibiza 2025-08-01 02:10:40 +05:30
parent 51aecd3e67
commit e1c1e19dee
2 changed files with 15 additions and 6 deletions

View File

@ -1933,9 +1933,9 @@ async def update_agent(
new_version_id = new_version.version_id new_version_id = new_version.version_id
update_data['current_version_id'] = new_version_id update_data['current_version_id'] = new_version_id
update_data['version_count'] = new_version['version_number'] update_data['version_count'] = new_version.version_number
logger.info(f"Created new version {new_version['version_name']} for agent {agent_id}") logger.info(f"Created new version {new_version.version_name} for agent {agent_id}")
except HTTPException: except HTTPException:
raise raise

View File

@ -12,9 +12,15 @@ from .api.agents import (
class Agent: class Agent:
def __init__(self, client: AgentsClient, agent_id: str): def __init__(
self,
client: AgentsClient,
agent_id: str,
model: str = "anthropic/claude-sonnet-4-20250514",
):
self._client = client self._client = client
self._agent_id = agent_id self._agent_id = agent_id
self._model = model
async def update( async def update(
self, self,
@ -49,18 +55,22 @@ class Agent:
), ),
) )
async def tool_details(self):
response = await self._client.get_agent(self._agent_id)
return response
async def run( async def run(
self, self,
prompt: str, prompt: str,
thread: Thread, thread: Thread,
model: str = "anthropic/claude-sonnet-4-20250514", model: str | None = None,
): ):
await thread.add_message(prompt) await thread.add_message(prompt)
response = await thread._client.start_agent( response = await thread._client.start_agent(
thread._thread_id, thread._thread_id,
AgentStartRequest( AgentStartRequest(
agent_id=self._agent_id, agent_id=self._agent_id,
model_name=model, model_name=model or self._model,
), ),
) )
return AgentRun(thread, response.agent_run_id) return AgentRun(thread, response.agent_run_id)
@ -74,7 +84,6 @@ class KortixAgent:
self, self,
name: str, name: str,
system_prompt: str, system_prompt: str,
model: str,
mcp_tools: list[KortixTools] = [], mcp_tools: list[KortixTools] = [],
) -> Agent: ) -> Agent:
agentpress_tools = {} agentpress_tools = {}