feat: pass threadId on agent build request (#30)

Author: samuelint

langchain_openai_api_bridge/core/create_agent_dto.py

@@ -8,3 +8,4 @@ class CreateAgentDto(BaseModel):
     temperature: float = 0.7
     max_tokens: Optional[int] = None
     assistant_id: Optional[str] = ""
+    thread_id: Optional[str] = ""

langchain_openai_api_bridge/fastapi/assistant_api_router.py

@@ -23,8 +23,9 @@
 from langchain_openai_api_bridge.assistant.repository.run_repository import (
     RunRepository,
 )
-from langchain_openai_api_bridge.core.agent_factory import AgentFactory
-from langchain_openai_api_bridge.core.create_agent_dto import CreateAgentDto
+from langchain_openai_api_bridge.fastapi.internal_agent_factory import (
+    InternalAgentFactory,
+)
 from langchain_openai_api_bridge.core.utils.tiny_di_container import TinyDIContainer
 from langchain_openai_api_bridge.fastapi.token_getter import get_bearer_token
 
@@ -117,16 +118,10 @@ async def assistant_create_thread_runs(
         thread_run_dto.thread_id = thread_id
 
         api_key = get_bearer_token(authorization)
-
-        agent_factory = tiny_di_container.resolve(AgentFactory)
-        create_agent_dto = CreateAgentDto(
-            model=thread_run_dto.model,
-            api_key=api_key,
-            temperature=thread_run_dto.temperature,
-            assistant_id=thread_run_dto.assistant_id,
+        agent_factory = tiny_di_container.resolve(InternalAgentFactory)
+        agent = agent_factory.create_agent(
+            thread_run_dto=thread_run_dto, api_key=api_key
         )
-        llm = agent_factory.create_llm(dto=create_agent_dto)
-        agent = agent_factory.create_agent(llm=llm, dto=create_agent_dto)
 
         service = tiny_di_container.resolve(AssistantRunService)
 

langchain_openai_api_bridge/fastapi/internal_agent_factory.py

@@ -0,0 +1,22 @@
+from langchain_openai_api_bridge.assistant.create_thread_runs_api_dto import (
+    ThreadRunsDto,
+)
+from langchain_openai_api_bridge.core.agent_factory import AgentFactory
+from langchain_openai_api_bridge.core.create_agent_dto import CreateAgentDto
+from langchain_core.runnables import Runnable
+
+
+class InternalAgentFactory:
+    def __init__(self, agent_factory: AgentFactory) -> None:
+        self.agent_factory = agent_factory
+
+    def create_agent(self, thread_run_dto: ThreadRunsDto, api_key: str) -> Runnable:
+        create_agent_dto = CreateAgentDto(
+            model=thread_run_dto.model,
+            thread_id=thread_run_dto.thread_id,
+            api_key=api_key,
+            temperature=thread_run_dto.temperature,
+            assistant_id=thread_run_dto.assistant_id,
+        )
+        llm = self.agent_factory.create_llm(dto=create_agent_dto)
+        return self.agent_factory.create_agent(llm=llm, dto=create_agent_dto)

langchain_openai_api_bridge/fastapi/langchain_openai_api_bridge_fastapi.py

@@ -32,6 +32,9 @@
 from langchain_openai_api_bridge.fastapi.chat_completion_router import (
     create_openai_chat_completion_router,
 )
+from langchain_openai_api_bridge.fastapi.internal_agent_factory import (
+    InternalAgentFactory,
+)
 
 
 class LangchainOpenaiApiBridgeFastAPI(LangchainOpenaiApiBridge):
@@ -79,6 +82,8 @@ def bind_openai_assistant_api(
         self.tiny_di_container.register(RunRepository, to=run_repository_provider)
         register_assistant_adapter(self.tiny_di_container)
 
+        self.tiny_di_container.register(InternalAgentFactory)
+
         assistant_router = create_openai_assistant_router(
             tiny_di_container=self.tiny_di_container,
             prefix=prefix,