Automated pre-commit update

Author: rushilpatel0

src/codegen/agents/__init__.py

@@ -1 +0,0 @@
-

src/codegen/agents/code_agent.py

@@ -7,8 +7,8 @@
 from langchain_core.runnables.config import RunnableConfig
 from langgraph.graph.graph import CompiledGraph
 from langsmith import Client
-from langchain_core.messages import AIMessage
-from codegen.agents.loggers import  ExternalLogger
+
+from codegen.agents.loggers import ExternalLogger
 from codegen.agents.tracer import MessageStreamTracer
 from codegen.extensions.langchain.agent import create_codebase_agent
 from codegen.extensions.langchain.utils.get_langsmith_url import (
@@ -86,7 +86,6 @@ def __init__(
         self.project_name = os.environ.get("LANGCHAIN_PROJECT", "RELACE")
         print(f"Using LangSmith project: {self.project_name}")
 
-
         # Store SWEBench metadata if provided
         self.run_id = metadata.get("run_id")
         self.instance_id = metadata.get("instance_id")
@@ -125,8 +124,6 @@ def run(self, prompt: str) -> str:
             "recursion_limit": 100,
         }
 
-
-
         # this message has a reducer which appends the current message to the existing history
         # see more https://langchain-ai.github.io/langgraph/concepts/low_level/#reducers
         input = {"query": prompt}
@@ -136,7 +133,6 @@ def run(self, prompt: str) -> str:
 
         stream = self.agent.stream(input, config=config, stream_mode="values")
 
-
         _tracer = MessageStreamTracer(logger=self.logger)
 
         # Process the stream with the tracer

src/codegen/agents/data.py

@@ -1,54 +1,71 @@
-from typing import List, Optional, Union
 from dataclasses import dataclass, field
 from datetime import datetime
+from typing import Optional, Union
+
 
 # Base dataclass for all message types
 @dataclass
 class BaseMessage:
     """Base class for all message types."""
+
     type: str
     timestamp: str = field(default_factory=lambda: datetime.now().isoformat())
     content: str = ""
 
+
 @dataclass
 class UserMessage(BaseMessage):
     """Represents a message from the user."""
+
     type: str = field(default="user")
 
+
 @dataclass
 class SystemMessageData(BaseMessage):
     """Represents a system message."""
+
     type: str = field(default="system")
 
+
 @dataclass
 class ToolCall:
     """Represents a tool call within an assistant message."""
+
     name: Optional[str] = None
     arguments: Optional[str] = None
     id: Optional[str] = None
 
+
 @dataclass
 class AssistantMessage(BaseMessage):
     """Represents a message from the assistant."""
+
     type: str = field(default="assistant")
-    tool_calls: List[ToolCall] = field(default_factory=list)
+    tool_calls: list[ToolCall] = field(default_factory=list)
+
 
 @dataclass
 class ToolMessageData(BaseMessage):
     """Represents a tool response message."""
+
     type: str = field(default="tool")
     tool_name: Optional[str] = None
     tool_response: Optional[str] = None
     tool_id: Optional[str] = None
 
+
 @dataclass
 class FunctionMessageData(BaseMessage):
     """Represents a function message."""
+
     type: str = field(default="function")
 
+
 @dataclass
 class UnknownMessage(BaseMessage):
     """Represents an unknown message type."""
-    type: str = field(default="unknown") 
 
-type AgentRunMessage = Union[UserMessage, SystemMessageData, AssistantMessage, ToolMessageData, FunctionMessageData, UnknownMessage]
+    type: str = field(default="unknown")
+
+
+type AgentRunMessage = Union[UserMessage, SystemMessageData, AssistantMessage, ToolMessageData, FunctionMessageData, UnknownMessage]

src/codegen/agents/loggers.py

@@ -1,15 +1,16 @@
 from typing import Protocol
+
 from .data import AgentRunMessage
 
+
 # Define the interface for ExternalLogger
 class ExternalLogger(Protocol):
     """Protocol defining the interface for external loggers."""
-    
+
     def log(self, data: AgentRunMessage) -> None:
-        """
-        Log structured data to an external system.
-        
+        """Log structured data to an external system.
+
         Args:
             data: The structured data to log, either as a dictionary or a BaseMessage
         """
-        pass
+        pass