diff --git a/llama_hub/tools/openai_image_generation/README.md b/llama_hub/tools/openai_image_generation/README.md index 04c1b39e18..35bf23214f 100644 --- a/llama_hub/tools/openai_image_generation/README.md +++ b/llama_hub/tools/openai_image_generation/README.md @@ -8,7 +8,7 @@ This tool has a more extensive example usage documented in a Jupyter notebook [h ### Usage with Agent ```python -from llama_hub.tools.openai.image_generation import OpenAIImageGenerationToolSpec +from llama_hub.tools.openai_image_generation import OpenAIImageGenerationToolSpec image_generation_tool = OpenAIImageGenerationToolSpec(api_key=os.environ["OPENAI_API_KEY"]) @@ -24,7 +24,7 @@ print(response) ### Usage directly ```python -from llama_hub.tools.openai.image_generation import OpenAIImageGenerationToolSpec +from llama_hub.tools.openai_image_generation import OpenAIImageGenerationToolSpec image_generation_tool = OpenAIImageGenerationToolSpec(api_key=os.environ["OPENAI_API_KEY"]) diff --git a/llama_hub/tools/openai_image_generation/__init__.py b/llama_hub/tools/openai_image_generation/__init__.py index fbcd8ef66a..3f05773510 100644 --- a/llama_hub/tools/openai_image_generation/__init__.py +++ b/llama_hub/tools/openai_image_generation/__init__.py @@ -1,5 +1,5 @@ ## init file -from llama_hub.tools.openai.image_generation.base import ( +from llama_hub.tools.openai_image_generation.base import ( OpenAIImageGenerationToolSpec, )