mirror of https://github.com/kortix-ai/suna.git
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:
parent
51aecd3e67
commit
e1c1e19dee
|
@ -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
|
||||
|
|
|
@ -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 = {}
|
||||
|
|
Loading…
Reference in New Issue