mirror of
https://github.com/openswarm-ai/openswarm.git
synced 2026-09-11 12:17:45 +02:00
[Haik]: fixed imports and added init py files. Now gonna add a couple sub folders in the pre_existing_toolkits, then move on to the session management
This commit is contained in:
@@ -56,7 +56,6 @@ class Agent(BaseModel):
|
||||
msg=msg,
|
||||
options=self.config,
|
||||
branch_id=self.branch_id,
|
||||
parent_id=self.parent_id,
|
||||
))
|
||||
|
||||
@typechecked
|
||||
|
||||
@@ -5,11 +5,11 @@ from claude_agent_sdk import (
|
||||
query, ClaudeAgentOptions, AssistantMessage,
|
||||
)
|
||||
from claude_agent_sdk.types import StreamEvent
|
||||
from backend.apps.agents.manager.HaikFix.run_agent_loop.helpers.handle_stream_event import handle_stream_event
|
||||
from backend.apps.agents.manager.HaikFix.run_agent_loop.helpers.handle_assistant_message import handle_assistant_message
|
||||
from backend.apps.HaikFix.Agent.run_agent_loop.helpers.handle_stream_event import handle_stream_event
|
||||
from backend.apps.HaikFix.Agent.run_agent_loop.helpers.handle_assistant_message import handle_assistant_message
|
||||
|
||||
# from backend.apps.agents.manager.HaikFix.shared_structs.Message.sub_types import ImageChunk, TextChunk
|
||||
from backend.apps.agents.manager.HaikFix.shared_structs.Message.Message import Message, PromptMsgDict
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.Message import Message, PromptMsgDict
|
||||
from typing import List, Dict, Literal, Union, Optional
|
||||
|
||||
@typechecked
|
||||
|
||||
@@ -37,7 +37,7 @@ class MessageLog(BaseModel):
|
||||
"""
|
||||
for i, m in enumerate[AnyMessage](self.messages):
|
||||
if m.id == message_id:
|
||||
return self._messages[:i + 1]
|
||||
return self.messages[:i + 1]
|
||||
raise ValueError(f"Message {message_id} not found")
|
||||
|
||||
@typechecked
|
||||
|
||||
@@ -24,7 +24,7 @@ class Toolkit(BaseModel):
|
||||
tools=tools,
|
||||
nested_toolkits=nested_toolkits
|
||||
)
|
||||
self.validate_structure(tools, nested_toolkits)
|
||||
self.validate_structure()
|
||||
|
||||
@typechecked
|
||||
def validate_structure(self) -> None:
|
||||
|
||||
Reference in New Issue
Block a user