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

2023-12-04 10:59:13 +07:00
parent e34b1e4c6d
commit 0ce3a8832f
34 changed files with 641 additions and 310 deletions
--- a/knowledgehub/agents/langchain.py
+++ b/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,