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"), ) ```