Hopefully fix the CI and tests

Author: pamelafox

.github/workflows/app-tests.yaml

@@ -48,6 +48,9 @@ jobs:
             brew install postgresql@14
             brew link --overwrite postgresql@14
             brew install pgvector
+            # pgvector installs into the default PostgreSQL, copy extension files to postgresql@14
+            cp /opt/homebrew/share/postgresql/extension/vector* /opt/homebrew/share/postgresql@14/extension/
+            cp /opt/homebrew/lib/postgresql/vector.dylib /opt/homebrew/lib/postgresql@14/vector.dylib 2>/dev/null || true
             brew services start postgresql@14 && sleep 1
             createuser -s ${{ env.POSTGRES_USERNAME }}
             psql -d postgres -c "ALTER USER ${{ env.POSTGRES_USERNAME }} WITH PASSWORD '${{ env.POSTGRES_PASSWORD }}'"

tests/conftest.py

@@ -13,6 +13,7 @@
 from openai.types.create_embedding_response import Usage
 from openai.types.responses import (
     Response,
+    ResponseCompletedEvent,
     ResponseFunctionToolCall,
     ResponseOutputMessage,
     ResponseOutputText,
@@ -212,13 +213,45 @@ def __init__(self, answer: str):
                         sequence_number=0,
                     )
                 )
+            # Agents SDK requires a ResponseCompletedEvent to finalize the stream
+            self.events.append(
+                ResponseCompletedEvent(
+                    type="response.completed",
+                    sequence_number=len(self.events),
+                    response=Response(
+                        id="resp-test-stream",
+                        created_at=0,
+                        model="gpt-5.4",
+                        object="response",
+                        output=[
+                            ResponseOutputMessage(
+                                id="msg-1",
+                                type="message",
+                                role="assistant",
+                                status="completed",
+                                content=[ResponseOutputText(type="output_text", text=answer, annotations=[])],
+                            )
+                        ],
+                        tool_choice="auto",
+                        tools=[],
+                        status="completed",
+                        parallel_tool_calls=True,
+                    ),
+                )
+            )
 
         async def __aenter__(self):
             return self
 
         async def __aexit__(self, exc_type, exc_val, exc_tb):
             return None
 
+        async def close(self):
+            pass
+
+        async def parse(self):
+            return self
+
         def __aiter__(self):
             return self
 

tests/snapshots/test_api_routes/test_advanced_chat_streaming_flow/advanced_chat_streaming_flow_response.jsonlines

@@ -1,2 +1,2 @@
 {"type":"response.context","delta":null,"context":{"data_points":{"1":{"id":1,"type":"Footwear","brand":"Daybird","name":"Wanderer Black Hiking Boots","description":"Daybird's Wanderer Hiking Boots in sleek black are perfect for all your outdoor adventures. These boots are made with a waterproof leather upper and a durable rubber sole for superior traction. With their cushioned insole and padded collar, these boots will keep you comfortable all day long.","price":109.99}},"thoughts":[{"title":"Prompt to generate search arguments","description":[{"role":"system","content":"Your job is to find search results based off the user's question and past messages.\nYou have access to only these tools:\n1. **search_database**: This tool allows you to search a table for items based on a query.\n  You can pass in a search query and optional filters.\nOnce you get the search results, you're done.\n"},{"role":"user","content":"good options for climbing gear that can be used outside?"},{"id":"fc_madeup1","call_id":"call_abc123","name":"search_database","arguments":"{\"search_query\":\"climbing gear outside\"}","type":"function_call"},{"id":"fc_madeupoutput1","call_id":"call_abc123","output":"Search results for climbing gear that can be used outside: ...","type":"function_call_output"},{"role":"user","content":"are there any shoes less than $50?"},{"id":"fc_madeup2","call_id":"call_abc456","name":"search_database","arguments":"{\"search_query\":\"shoes\",\"price_filter\":{\"comparison_operator\":\"<\",\"value\":50}}","type":"function_call"},{"id":"fc_madeupoutput2","call_id":"call_abc456","output":"Search results for shoes cheaper than 50: ...","type":"function_call_output"},{"role":"user","content":"Find search results for user query: What is the capital of France?"}],"props":{"model":"gpt-5.4","deployment":"gpt-5.4"}},{"title":"Search using generated search arguments","description":"climbing gear outside","props":{"top":1,"vector_search":true,"text_search":true,"filters":[]}},{"title":"Search results","description":[{"id":1,"type":"Footwear","brand":"Daybird","name":"Wanderer Black Hiking Boots","description":"Daybird's Wanderer Hiking Boots in sleek black are perfect for all your outdoor adventures. These boots are made with a waterproof leather upper and a durable rubber sole for superior traction. With their cushioned insole and padded collar, these boots will keep you comfortable all day long.","price":109.99}],"props":{}},{"title":"Prompt to generate answer","description":[{"role":"system","content":"Assistant helps customers with questions about products.\nRespond as if you are a salesperson helping a customer in a store. Do NOT respond with tables.\nAnswer ONLY with the product details listed in the products.\nIf there isn't enough information below, say you don't know.\nDo not generate answers that don't use the sources below.\nEach product has an ID in brackets followed by colon and the product details.\nAlways include the product ID for each product you use in the response.\nUse square brackets to reference the source, for example [52].\nDon't combine citations, list each product separately, for example [27][51]."},{"content":"What is the capital of France?Sources:\n[1]:Name:Wanderer Black Hiking Boots Description:Daybird's Wanderer Hiking Boots in sleek black are perfect for all your outdoor adventures. These boots are made with a waterproof leather upper and a durable rubber sole for superior traction. With their cushioned insole and padded collar, these boots will keep you comfortable all day long. Price:109.99 Brand:Daybird Type:Footwear","role":"user"}],"props":{"model":"gpt-5.4","deployment":"gpt-5.4"}}]}}
-{"error": "'AsyncResponseEventIterator' object has no attribute 'close'"}
+{"type":"response.output_text.delta","delta":"The capital of France is Paris. [Benefit_Options-2.pdf].","context":null}

tests/snapshots/test_api_routes/test_simple_chat_streaming_flow/simple_chat_streaming_flow_response.jsonlines

@@ -1,2 +1,2 @@
 {"type":"response.context","delta":null,"context":{"data_points":{"1":{"id":1,"type":"Footwear","brand":"Daybird","name":"Wanderer Black Hiking Boots","description":"Daybird's Wanderer Hiking Boots in sleek black are perfect for all your outdoor adventures. These boots are made with a waterproof leather upper and a durable rubber sole for superior traction. With their cushioned insole and padded collar, these boots will keep you comfortable all day long.","price":109.99}},"thoughts":[{"title":"Search query for database","description":"What is the capital of France?","props":{"top":1,"vector_search":true,"text_search":true}},{"title":"Search results","description":[{"id":1,"type":"Footwear","brand":"Daybird","name":"Wanderer Black Hiking Boots","description":"Daybird's Wanderer Hiking Boots in sleek black are perfect for all your outdoor adventures. These boots are made with a waterproof leather upper and a durable rubber sole for superior traction. With their cushioned insole and padded collar, these boots will keep you comfortable all day long.","price":109.99}],"props":{}},{"title":"Prompt to generate answer","description":[{"role":"system","content":"Assistant helps customers with questions about products.\nRespond as if you are a salesperson helping a customer in a store. Do NOT respond with tables.\nAnswer ONLY with the product details listed in the products.\nIf there isn't enough information below, say you don't know.\nDo not generate answers that don't use the sources below.\nEach product has an ID in brackets followed by colon and the product details.\nAlways include the product ID for each product you use in the response.\nUse square brackets to reference the source, for example [52].\nDon't combine citations, list each product separately, for example [27][51]."},{"content":"What is the capital of France?Sources:\n[1]:Name:Wanderer Black Hiking Boots Description:Daybird's Wanderer Hiking Boots in sleek black are perfect for all your outdoor adventures. These boots are made with a waterproof leather upper and a durable rubber sole for superior traction. With their cushioned insole and padded collar, these boots will keep you comfortable all day long. Price:109.99 Brand:Daybird Type:Footwear","role":"user"}],"props":{"model":"gpt-5.4","deployment":"gpt-5.4"}}]}}
-{"error": "'AsyncResponseEventIterator' object has no attribute 'close'"}
+{"type":"response.output_text.delta","delta":"The capital of France is Paris. [Benefit_Options-2.pdf].","context":null}