Add new functions for image generation with DALL-E 3

rayisaninja · Jan 7, 2024 · f375b53 · f375b53
1 parent 238e637
commit f375b53
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 1 deletion.
diff --git a/app.py b/app.py
@@ -390,6 +390,13 @@ async def main():
 
     # Define core tools here
     tools = [
+        {
+            "type": "function",
+            "function": {
+                "name": "get_current_date_time",
+                "description": "Get the current date and time from the local machine.",
+            },
+        },
         {
             "type": "function",
             "function": {
@@ -507,6 +514,43 @@ async def main():
                 },
             },
         },
+        {
+            "type": "function",
+            "function": {
+                "name": "generate_an_image_with_dalle3",
+                "description": "Generate an image with DALL-E 3.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "prompt": {
+                            "type": "string",
+                            "description": "The prompt to use for image generation.",
+                        },
+                        "n": {
+                            "type": "integer",
+                            "description": "The number of images to generate.",
+                        },
+                        "size": {
+                            "type": "string",
+                            "description": "The image size to generate.",
+                        },
+                        "quality": {
+                            "type": "string",
+                            "description": "The image quality to generate.",
+                        },
+                        "style": {
+                            "type": "string",
+                            "description": "The image style to generate.",
+                        },
+                        "response_format": {
+                            "type": "string",
+                            "description": "The response format to use for image generation.",
+                        },
+                    },
+                    "required": ["prompt"],
+                },
+            },
+        },
     ]
 
     # Use the load_plugins_and_get_tools function to conditionally add tools

diff --git a/utils/openai_dalle_tools.py b/utils/openai_dalle_tools.py
@@ -18,7 +18,7 @@
 openai_org_id = OPENAI_ORG_ID
 
 # Create an OpenAI client instance using keyword arguments
-client = OpenAI(api_key=api_key, organization=openai_org_id, timeout=10)
+client = OpenAI(api_key=api_key, organization=openai_org_id, timeout=60)
 
 # Create an AsyncOpenAI client instance using keyword arguments
 client_async = AsyncOpenAI(api_key=api_key, organization=openai_org_id, timeout=60)