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
|
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
|
||||||
|
|
|
@ -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 = {}
|
||||||
|
|
Loading…
Reference in New Issue