enrichment with years, events

services/rag/langchain/agent.py

@@ -21,7 +21,9 @@ from vector_storage import initialize_vector_store
 load_dotenv()
 
 
-def get_llm_model_info(llm_model: str = None) -> Tuple[str, str, str, str, str]:
+def get_llm_model_info(
+    llm_model: Optional[str] = None,
+) -> Tuple[str, str, str, str, str]:
     """
     Get LLM model information based on environment configuration.
 
@@ -121,7 +123,7 @@ class DocumentRetrievalTool(BaseTool):
 
 
 def create_chat_agent(
-    collection_name: str = "documents_langchain", llm_model: str = None
+    collection_name: str = "documents_langchain", llm_model: Optional[str] = None
 ) -> Any:
     """
     Create a chat agent with document retrieval capabilities.
@@ -177,7 +179,7 @@ def create_chat_agent(
 def chat_with_agent(
     query: str,
     collection_name: str = "documents_langchain",
-    llm_model: str = None,
+    llm_model: Optional[str] = None,
     history: List[BaseMessage] = None,
 ) -> Dict[str, Any]:
     """