[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:
haikdc
2026-04-02 16:30:12 -07:00
parent df07f82253
commit 12931443fb
22 changed files with 5 additions and 6 deletions
-1
View File
@@ -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: