Fix save

run-llama · Feb 16, 2024 · 2e7ba38 · 2e7ba38
1 parent 4fc3cb9
commit 2e7ba38
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/llama-index-integrations/agent/llama-index-agent-openai/llama_index/agent/openai/step.py b/llama-index-integrations/agent/llama-index-agent-openai/llama_index/agent/openai/step.py
@@ -80,7 +80,7 @@ def call_function(
     tools: List[BaseTool],
     tool_call: OpenAIToolCall,
     verbose: bool = False,
-    get_tool_by: Calla
+    get_tool_by: Callable[[List[BaseTool], str], BaseTool] = get_function_by_name,
 ) -> Tuple[ChatMessage, ToolOutput]:
     """Call a function and return the output as a string."""
     # validations to get passed mypy

diff --git a/pyproject.toml b/pyproject.toml
@@ -92,7 +92,6 @@ llamaindex-cli = 'llama_index.core.command_line.command_line:main'
 
 [[tool.poetry.source]]
 name = "nvidia-pypi"
-priority = "supplemental"
 url = "https://pypi.nvidia.com"
 
 [tool.ruff]