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
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:
raise

View File

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