Update tools/describe_image_tool.py
Browse files
tools/describe_image_tool.py
CHANGED
|
@@ -4,7 +4,15 @@ import os
|
|
| 4 |
from openai import OpenAI
|
| 5 |
from smolagents import Tool
|
| 6 |
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
class DescribeImageTool(Tool):
|
|
|
|
| 4 |
from openai import OpenAI
|
| 5 |
from smolagents import Tool
|
| 6 |
|
| 7 |
+
import os
|
| 8 |
+
from smolagents import AzureOpenAIServerModel
|
| 9 |
+
|
| 10 |
+
model = AzureOpenAIServerModel(
|
| 11 |
+
model_id = os.environ.get("AZURE_OPENAI_MODEL"),
|
| 12 |
+
azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT"),
|
| 13 |
+
api_key=os.environ.get("AZURE_OPENAI_API_KEY"),
|
| 14 |
+
api_version=os.environ.get("OPENAI_API_VERSION")
|
| 15 |
+
)
|
| 16 |
|
| 17 |
|
| 18 |
class DescribeImageTool(Tool):
|