fix: parallel tests

langchain_openai_api_bridge/core/langchain_openai_api_bridge.py

@@ -1,14 +1,14 @@
-from typing import Callable
+from typing import Callable, Optional, Union
 from langchain_openai_api_bridge.core.agent_factory import AgentFactory
 from langchain_openai_api_bridge.core.utils import TinyDIContainer
 
 
 class LangchainOpenaiApiBridge:
     def __init__(
         self,
-        agent_factory_provider: Callable[[], AgentFactory],
-        tiny_di_container: TinyDIContainer = TinyDIContainer(),
+        agent_factory_provider: Union[Callable[[], AgentFactory], AgentFactory],
+        tiny_di_container: Optional[TinyDIContainer] = None,
     ) -> None:
-        self.tiny_di_container = tiny_di_container
+        self.tiny_di_container = tiny_di_container or TinyDIContainer()
 
         self.tiny_di_container.register(AgentFactory, to=agent_factory_provider)

langchain_openai_api_bridge/core/utils/tiny_di_container.py

@@ -1,7 +1,6 @@
 from typing import Type, TypeVar, Dict, Any, Union, Callable
 import inspect
 
-
 T = TypeVar("T")
 
 
@@ -25,6 +24,8 @@ def register(
         else:
             self.services[cls] = service if service else implementation
 
+        self.implementations[cls] = implementation
+
     def resolve(self, cls: Type[T]) -> T:
         if cls in self.singletons:
             if isinstance(self.singletons[cls], type):

langchain_openai_api_bridge/fastapi/langchain_openai_api_bridge_fastapi.py

@@ -1,4 +1,4 @@
-from typing import Callable
+from typing import Callable, Union
 from fastapi import FastAPI
 
 from langchain_openai_api_bridge.assistant.adapter.container import (
@@ -36,16 +36,18 @@
 
 class LangchainOpenaiApiBridgeFastAPI(LangchainOpenaiApiBridge):
     def __init__(
-        self, app: FastAPI, agent_factory_provider: Callable[[], AgentFactory]
+        self,
+        app: FastAPI,
+        agent_factory_provider: Union[Callable[[], AgentFactory], AgentFactory],
     ) -> None:
         super().__init__(agent_factory_provider=agent_factory_provider)
         self.app = app
 
     def bind_openai_assistant_api(
         self,
-        thread_repository_provider: Callable[[], ThreadRepository],
-        message_repository_provider: Callable[[], MessageRepository],
-        run_repository_provider: Callable[[], RunRepository],
+        thread_repository_provider: Union[Callable[[], ThreadRepository]],
+        message_repository_provider: Union[Callable[[], MessageRepository]],
+        run_repository_provider: Union[Callable[[], RunRepository]],
         prefix: str = "",
     ) -> None: