[adding test script]

Author: aangelopoulos

fastchat/serve/scripts/test.py

@@ -0,0 +1,26 @@
+import os
+from litellm import image_generation
+
+# Set up API keys and credentials
+# export OPENAI_API_KEY = "your_openai_api_key"
+# export VERTEX_AI_PROJECT = "your_vertex_project_id"
+# export VERTEX_AI_LOCATION = "us-central1"
+
+prompt = "A cute baby sea otter"
+
+# OpenAI
+
+#response = image_generation(prompt=prompt, model="dall-e-3")
+
+#print(f"response: {response.url}")
+
+# Vertex AI
+
+response = image_generation(
+    prompt=prompt,
+    model="vertex_ai/imagegeneration@006",
+    vertex_ai_project="adroit-crow-413218",
+    vertex_ai_location="us-central1",
+)
+
+print(f"response: {response}")