fix: errors

ComposioHQ · Mar 24, 2024 · 1eeb43b · 1eeb43b
1 parent 866fb7f
commit 1eeb43b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/langchain/composio_langchain/composio_tool_spec.py b/langchain/composio_langchain/composio_tool_spec.py
@@ -26,7 +26,8 @@ def pydantic_model_from_param_schema(param_schema):
     fields = {}
     param_title = param_schema['title'].replace(" ", "")
     required_props = param_schema.get('required', [])
-    for prop_name, prop_info in param_schema['properties'].items():
+    schema_params_object = param_schema.get('properties', {})
+    for prop_name, prop_info in schema_params_object.items():
         prop_type = prop_info["type"]
         prop_title = prop_info['title'].replace(" ", "")
         prop_default = prop_info.get('default', fallback_values[prop_type])
@@ -56,8 +57,8 @@ def get_signature_format_from_schema_params(
 ):
     parameters = []
     required_params = schema_params.get('required', [])
-    schema_parms_object = schema_params.get('properties', {})
-    for param_name, param_schema in schema_parms_object.items():
+    schema_params_object = schema_params.get('properties', {})
+    for param_name, param_schema in schema_params_object.items():
         param_type = param_schema['type']
         param_title = param_schema['title'].replace(" ", "")
 

diff --git a/langchain/test.py b/langchain/test.py
@@ -1,12 +1,10 @@
 from langchain.agents import create_openai_functions_agent
 from langchain.agents import AgentExecutor
 from langchain import hub
-from composio_langchain import ComposioToolset
+from composio_langchain import ComposioToolset, App
 from langchain_openai import ChatOpenAI
-from composio.sdk import App
 
 llm = ChatOpenAI(openai_api_key="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")
-
 prompt = hub.pull("hwchase17/openai-functions-agent")
 
 tools = ComposioToolset([App.GITHUB])