From d140e98f021596f3087fec6ca52b174387947fbe Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Sun, 9 Jun 2024 13:47:43 -0700 Subject: [PATCH] fix docs (#122) --- README.md | 3 ++- docs/index.md | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index f9c66156..ca290998 100644 --- a/README.md +++ b/README.md @@ -160,11 +160,12 @@ You can then run Vision Agent using the Azure OpenAI models: ```python import vision_agent as va +import vision_agent.tools as T agent = va.agent.VisionAgent( planner=va.llm.AzureOpenAILLM(), coder=va.lmm.AzureOpenAILMM(), tester=va.lmm.AzureOpenAILMM(), debugger=va.lmm.AzureOpenAILMM(), - tool_recommender=va.utils.AzureSim(), + tool_recommender=va.utils.AzureSim(T.TOOLS_DF, sim_key="desc"), ) ``` diff --git a/docs/index.md b/docs/index.md index dc174285..4225ff17 100644 --- a/docs/index.md +++ b/docs/index.md @@ -150,11 +150,12 @@ You can then run Vision Agent using the Azure OpenAI models: ```python import vision_agent as va +import vision_agent.tools as T agent = va.agent.VisionAgent( planner=va.llm.AzureOpenAILLM(), coder=va.lmm.AzureOpenAILMM(), tester=va.lmm.AzureOpenAILMM(), debugger=va.lmm.AzureOpenAILMM(), - tool_recommender=va.utils.AzureSim(), + tool_recommender=va.utils.AzureSim(T.TOOLS_DF, sim_key="desc"), ) ```