Provide type hints for pass-through Langchain and Llama-index objects (#95)

knowledgehub/agents/langchain.py

@@ -54,7 +54,7 @@ class LangchainAgent(BaseAgent):
             # reinit Langchain AgentExecutor
             self.agent = initialize_agent(
                 langchain_plugins,
-                self.llm.agent,
+                self.llm._obj,
                 agent=self.AGENT_TYPE_MAP[self.agent_type],
                 handle_parsing_errors=True,
                 verbose=True,

knowledgehub/agents/tools/llm.py

@@ -21,7 +21,7 @@ class LLMTool(BaseTool):
         "are confident in solving the problem "
         "yourself. Input can be any instruction."
     )
-    llm: BaseLLM = AzureChatOpenAI()
+    llm: BaseLLM = AzureChatOpenAI.withx()
     args_schema: Optional[Type[BaseModel]] = LLMArgs
 
     def _run_tool(self, query: AnyStr) -> str: