Fix typo

databricks · Jan 15, 2025 · 8200ab0 · 8200ab0
1 parent 25da349
commit 8200ab0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/openai/src/databricks_openai/vector_search_retriever_tool.py b/integrations/openai/src/databricks_openai/vector_search_retriever_tool.py
@@ -31,7 +31,7 @@ class VectorSearchRetrieverTool(VectorSearchRetrieverToolMixin):
             messages=initial_messages,
             tools=tools,
         )
-        tool_call = chat_completion_resp.choices[0].message.tool_calls[0]
+        tool_call = response.choices[0].message.tool_calls[0]
         args = json.loads(tool_call.function.arguments)
         retriever_resp = vector_search_tool.execute(query=args["query"])
     """