Added type hint

chriscarrollsmith · chriscarrollsmith · commit 17e8ddc50a26 · 2024-11-02T15:28:21.000-04:00
diff --git a/routers/messages.py b/routers/messages.py
@@ -51,7 +51,7 @@ async def stream_response(
     # Create a generator to stream the response from the assistant
     # Create a generator to stream the response from the assistant
     async def event_generator():
-        stream = openai.beta.threads.runs.stream(
+        stream: AsyncAssistantStreamManager = openai.beta.threads.runs.stream(
             assistant_id=assistant_id,
             thread_id=thread_id
         )

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ async def stream_response(`
`51`	`51`	`# Create a generator to stream the response from the assistant`
`52`	`52`	`# Create a generator to stream the response from the assistant`
`53`	`53`	`async def event_generator():`
`54`		`- stream = openai.beta.threads.runs.stream(`
	`54`	`+ stream: AsyncAssistantStreamManager = openai.beta.threads.runs.stream(`
`55`	`55`	`assistant_id=assistant_id,`
`56`	`56`	`thread_id=thread_id`
`57`	`57`	`)`