fix tools/openai_image_generation import (#906)

* fix tools/openai_image_generation import * fix tools/openai_image_generation/README.md example code import
run-llama · Jan 29, 2024 · 8bb1882 · 8bb1882
1 parent e382a46
commit 8bb1882
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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,
 )