Set up strongly typed create_assistant script

Author: chriscarrollsmith

README.md

@@ -26,13 +26,19 @@ uv venv
 uv pip install -r pyproject.toml
 ```
 
-### 4. Run
+### 4. Create an assistant
+
+```shell
+uv run create_assistant.py
+```
+
+### 5. Run the FastAPI server
 
 ```shell
 uv run uvicorn main:app --host 0.0.0.0 --port 8000 --reload
 ```
 
-### 5. Navigate to [http://localhost:3000](http://localhost:3000).
+### 6. Navigate to [http://localhost:3000](http://localhost:3000).
 
 ## Overview
 

create_assistant.py

@@ -0,0 +1,60 @@
+import asyncio
+from dotenv import load_dotenv
+from openai import AsyncOpenAI
+from openai.types.beta.assistant_create_params import AssistantCreateParams
+from openai.types.beta.assistant_tool_param import CodeInterpreterToolParam, FileSearchToolParam, FunctionToolParam
+from openai.types.beta.assistant import Assistant
+from openai.types import FunctionDefinition
+
+request: AssistantCreateParams = AssistantCreateParams(
+    instructions="You are a helpful assistant.",
+    name="Quickstart Assistant",
+    model="gpt-4o",
+    tools=[
+        CodeInterpreterToolParam(type="code_interpreter"),
+        FileSearchToolParam(type="file_search"),
+        FunctionToolParam(
+            type="function",
+            function=FunctionDefinition(
+                name="get_weather",
+                description="Determine weather in my location",
+                parameters={
+                    "type": "object",
+                    "properties": {
+                        "location": {
+                            "type": "string",
+                            "description": "The city and state e.g. San Francisco, CA"
+                        }
+                    },
+                    "required": ["location"],
+                    "additionalProperties": False,
+                },
+                strict=True,
+            )
+        ),
+    ],
+)
+
+
+async def main():
+    # Create a new assistant using the OpenAI client
+    assistant: Assistant = await openai.beta.assistants.create(**request)
+    return assistant
+
+
+# Run the assistant creation in an asyncio event loop
+if __name__ == "__main__":
+    import logging
+    import sys
+
+    # Initialize the OpenAI client
+    load_dotenv()
+    openai: AsyncOpenAI = AsyncOpenAI()
+
+    # Configure logger to stream to console
+    logging.basicConfig(level=logging.INFO, stream=sys.stdout)
+    logger: logging.Logger = logging.getLogger(__name__)
+
+    # Run the main function in an asyncio event loop
+    assistant: Assistant = asyncio.run(main())
+    logger.info(f"Assistant created with ID: {assistant.id}")

routes/assistants.py

@@ -1,18 +1,21 @@
+from dotenv import load_dotenv
 from fastapi import FastAPI, Request, UploadFile, File
 from pydantic import BaseModel
 from openai import AsyncOpenAI
 
+load_dotenv()
+
 app = FastAPI()
 
 # Initialize OpenAI client
-openai = AsyncOpenAI(api_key="your-api-key")
+openai = AsyncOpenAI()
 
 # Pydantic model for DELETE request body
 class DeleteRequest(BaseModel):
     fileId: str
 
 # Helper function to get or create a vector store
-async def get_or_create_vector_store() -> str:
+async def get_or_create_vector_store(assistantId: str) -> str:
     assistant = await openai.beta.assistants.retrieve(assistantId)
     if assistant.tool_resources and assistant.tool_resources.file_search and assistant.tool_resources.file_search.vector_store_ids:
         return assistant.tool_resources.file_search.vector_store_ids[0]