diff --git a/tests/test_helpers/cassettes/test_assistants/test_AIAssistant_with_rag_invoke.yaml b/tests/test_helpers/cassettes/test_assistants/test_AIAssistant_with_rag_invoke.yaml index 9bf5ca3..15e09ec 100644 --- a/tests/test_helpers/cassettes/test_assistants/test_AIAssistant_with_rag_invoke.yaml +++ b/tests/test_helpers/cassettes/test_assistants/test_AIAssistant_with_rag_invoke.yaml @@ -32,18 +32,20 @@ interactions: method: POST uri: https://api.openai.com/v1/chat/completions response: - content: "{\n \"id\": \"chatcmpl-AEgzDhivTNpFojxM1BuqTCzQRnzye\",\n \"object\": - \"chat.completion\",\n \"created\": 1728065195,\n \"model\": \"gpt-4o-2024-08-06\",\n + content: "{\n \"id\": \"chatcmpl-AVJ9q7AF8QkzAVEYupnQ2tna2f3eD\",\n \"object\": + \"chat.completion\",\n \"created\": 1732025174,\n \"model\": \"gpt-4o-2024-08-06\",\n \ \"choices\": [\n {\n \"index\": 0,\n \"message\": {\n \"role\": - \"assistant\",\n \"content\": \"You're right by the American Museum of - Natural History, a fascinating place with exhibits on everything from dinosaurs - to space. Enjoy a stroll through Central Park, which offers beautiful landscapes, - walking paths, and iconic spots like Bow Bridge and Bethesda Terrace.\",\n \"refusal\": - null\n },\n \"logprobs\": null,\n \"finish_reason\": \"stop\"\n - \ }\n ],\n \"usage\": {\n \"prompt_tokens\": 124,\n \"completion_tokens\": - 47,\n \"total_tokens\": 171,\n \"prompt_tokens_details\": {\n \"cached_tokens\": - 0\n },\n \"completion_tokens_details\": {\n \"reasoning_tokens\": - 0\n }\n },\n \"system_fingerprint\": \"fp_2f406b9113\"\n}\n" + \"assistant\",\n \"content\": \"You are right by the American Museum + of Natural History, where you can explore fascinating exhibits on dinosaurs, + space, and human cultures. In addition, Central Park offers scenic walking paths, + boating on the lake, and the iconic Bethesda Terrace. Enjoy your visit!\",\n + \ \"refusal\": null\n },\n \"logprobs\": null,\n \"finish_reason\": + \"stop\"\n }\n ],\n \"usage\": {\n \"prompt_tokens\": 124,\n \"completion_tokens\": + 51,\n \"total_tokens\": 175,\n \"prompt_tokens_details\": {\n \"cached_tokens\": + 0,\n \"audio_tokens\": 0\n },\n \"completion_tokens_details\": {\n + \ \"reasoning_tokens\": 0,\n \"audio_tokens\": 0,\n \"accepted_prediction_tokens\": + 0,\n \"rejected_prediction_tokens\": 0\n }\n },\n \"system_fingerprint\": + \"fp_45cf54deae\"\n}\n" headers: Connection: - keep-alive @@ -67,13 +69,12 @@ interactions: which can be understood without the chat history. Do NOT answer the question, just reformulate it if needed and otherwise return it as is.", "role": "system"}, {"content": "I''m at Central Park W & 79st, New York, NY 10024, United States.", - "role": "user"}, {"content": "You''re right by the American Museum of Natural - History, a fascinating place with exhibits on everything from dinosaurs to space. - Enjoy a stroll through Central Park, which offers beautiful landscapes, walking - paths, and iconic spots like Bow Bridge and Bethesda Terrace.", "role": "assistant"}, - {"content": "content=''11 W 53rd St, New York, NY 10019, United States.'' additional_kwargs={} - response_metadata={} id=''11''", "role": "user"}], "model": "gpt-4o", "n": 1, - "stream": false, "temperature": 1.0}' + "role": "user"}, {"content": "You are right by the American Museum of Natural + History, where you can explore fascinating exhibits on dinosaurs, space, and + human cultures. In addition, Central Park offers scenic walking paths, boating + on the lake, and the iconic Bethesda Terrace. Enjoy your visit!", "role": "assistant"}, + {"content": "11 W 53rd St, New York, NY 10019, United States.", "role": "user"}], + "model": "gpt-4o", "n": 1, "stream": false, "temperature": 1.0}' headers: accept: - application/json @@ -84,7 +85,7 @@ interactions: connection: - keep-alive content-length: - - '930' + - '867' content-type: - application/json host: @@ -96,16 +97,18 @@ interactions: method: POST uri: https://api.openai.com/v1/chat/completions response: - content: "{\n \"id\": \"chatcmpl-AEgzFbJHJ9XNvxQPetmr4PZNphaYq\",\n \"object\": - \"chat.completion\",\n \"created\": 1728065197,\n \"model\": \"gpt-4o-2024-08-06\",\n + content: "{\n \"id\": \"chatcmpl-AVJ9sgunfHk3xWTpzuIALwmYJ1f48\",\n \"object\": + \"chat.completion\",\n \"created\": 1732025176,\n \"model\": \"gpt-4o-2024-08-06\",\n \ \"choices\": [\n {\n \"index\": 0,\n \"message\": {\n \"role\": - \"assistant\",\n \"content\": \"I see you're mentioning a new location. - Are you planning to visit the Museum of Modern Art at this address?\",\n \"refusal\": - null\n },\n \"logprobs\": null,\n \"finish_reason\": \"stop\"\n - \ }\n ],\n \"usage\": {\n \"prompt_tokens\": 167,\n \"completion_tokens\": - 22,\n \"total_tokens\": 189,\n \"prompt_tokens_details\": {\n \"cached_tokens\": - 0\n },\n \"completion_tokens_details\": {\n \"reasoning_tokens\": - 0\n }\n },\n \"system_fingerprint\": \"fp_e5e4913e83\"\n}\n" + \"assistant\",\n \"content\": \"The address you mentioned is for the + Museum of Modern Art (MoMA). What are you specifically looking for or interested + in at this location?\",\n \"refusal\": null\n },\n \"logprobs\": + null,\n \"finish_reason\": \"stop\"\n }\n ],\n \"usage\": {\n \"prompt_tokens\": + 159,\n \"completion_tokens\": 28,\n \"total_tokens\": 187,\n \"prompt_tokens_details\": + {\n \"cached_tokens\": 0,\n \"audio_tokens\": 0\n },\n \"completion_tokens_details\": + {\n \"reasoning_tokens\": 0,\n \"audio_tokens\": 0,\n \"accepted_prediction_tokens\": + 0,\n \"rejected_prediction_tokens\": 0\n }\n },\n \"system_fingerprint\": + \"fp_467dc6e35f\"\n}\n" headers: Connection: - keep-alive @@ -132,10 +135,10 @@ interactions: maximum and keep your suggestions concise.\n\n---START OF CONTEXT---\nCentral Park\n\nAmerican Museum of Natural History---END OF CONTEXT---\n\n", "role": "system"}, {"content": "I''m at Central Park W & 79st, New York, NY 10024, United - States.", "role": "user"}, {"content": "You''re right by the American Museum - of Natural History, a fascinating place with exhibits on everything from dinosaurs - to space. Enjoy a stroll through Central Park, which offers beautiful landscapes, - walking paths, and iconic spots like Bow Bridge and Bethesda Terrace.", "role": + States.", "role": "user"}, {"content": "You are right by the American Museum + of Natural History, where you can explore fascinating exhibits on dinosaurs, + space, and human cultures. In addition, Central Park offers scenic walking paths, + boating on the lake, and the iconic Bethesda Terrace. Enjoy your visit!", "role": "assistant"}, {"content": "11 W 53rd St, New York, NY 10019, United States.", "role": "user"}], "model": "gpt-4o", "n": 1, "stream": false, "temperature": 1.0}' @@ -149,7 +152,7 @@ interactions: connection: - keep-alive content-length: - - '1105' + - '1102' content-type: - application/json host: @@ -161,18 +164,20 @@ interactions: method: POST uri: https://api.openai.com/v1/chat/completions response: - content: "{\n \"id\": \"chatcmpl-AEgzGbPWcNf9UC78PouyK4Jdo0KoM\",\n \"object\": - \"chat.completion\",\n \"created\": 1728065198,\n \"model\": \"gpt-4o-2024-08-06\",\n + content: "{\n \"id\": \"chatcmpl-AVJ9sOdtOnJMts2zKtyoOJVjK9VlC\",\n \"object\": + \"chat.completion\",\n \"created\": 1732025176,\n \"model\": \"gpt-4o-2024-08-06\",\n \ \"choices\": [\n {\n \"index\": 0,\n \"message\": {\n \"role\": - \"assistant\",\n \"content\": \"You're near the Museum of Modern Art - (MoMA), home to an impressive collection of modern and contemporary artworks. - Just a short walk away, you'll find Rockefeller Center, offering shopping, dining, - and an observation deck with views of the city.\",\n \"refusal\": null\n - \ },\n \"logprobs\": null,\n \"finish_reason\": \"stop\"\n }\n - \ ],\n \"usage\": {\n \"prompt_tokens\": 197,\n \"completion_tokens\": - 48,\n \"total_tokens\": 245,\n \"prompt_tokens_details\": {\n \"cached_tokens\": - 0\n },\n \"completion_tokens_details\": {\n \"reasoning_tokens\": - 0\n }\n },\n \"system_fingerprint\": \"fp_2f406b9113\"\n}\n" + \"assistant\",\n \"content\": \"You're at the Museum of Modern Art, which + features an impressive collection of contemporary and modern artworks, including + pieces by Van Gogh and Warhol. Nearby, you can also visit the iconic Rockefeller + Center. Enjoy the artistic and cultural experiences!\",\n \"refusal\": + null\n },\n \"logprobs\": null,\n \"finish_reason\": \"stop\"\n + \ }\n ],\n \"usage\": {\n \"prompt_tokens\": 201,\n \"completion_tokens\": + 46,\n \"total_tokens\": 247,\n \"prompt_tokens_details\": {\n \"cached_tokens\": + 0,\n \"audio_tokens\": 0\n },\n \"completion_tokens_details\": {\n + \ \"reasoning_tokens\": 0,\n \"audio_tokens\": 0,\n \"accepted_prediction_tokens\": + 0,\n \"rejected_prediction_tokens\": 0\n }\n },\n \"system_fingerprint\": + \"fp_45cf54deae\"\n}\n" headers: Connection: - keep-alive diff --git a/tests/test_helpers/test_assistants.py b/tests/test_helpers/test_assistants.py index 5cb0bb0..c696c70 100644 --- a/tests/test_helpers/test_assistants.py +++ b/tests/test_helpers/test_assistants.py @@ -206,17 +206,17 @@ def test_AIAssistant_with_rag_invoke(): assert response_0["input"] == "I'm at Central Park W & 79st, New York, NY 10024, United States." assert response_0["output"] == ( - "You're right by the American Museum of Natural History, " - "a fascinating place with exhibits on everything from dinosaurs to space. " - "Enjoy a stroll through Central Park, which offers beautiful landscapes, " - "walking paths, and iconic spots like Bow Bridge and Bethesda Terrace." + "You are right by the American Museum of Natural History, where you can explore " + "fascinating exhibits on dinosaurs, space, and human cultures. " + "In addition, Central Park offers scenic walking paths, boating on the lake, " + "and the iconic Bethesda Terrace. Enjoy your visit!" ) assert response_1["input"] == "11 W 53rd St, New York, NY 10019, United States." assert response_1["output"] == ( - "You're near the Museum of Modern Art (MoMA), " - "home to an impressive collection of modern and contemporary artworks. " - "Just a short walk away, you'll find Rockefeller Center, offering shopping, " - "dining, and an observation deck with views of the city." + "You're at the Museum of Modern Art, which features an impressive collection of " + "contemporary and modern artworks, including pieces by Van Gogh and Warhol. " + "Nearby, you can also visit the iconic Rockefeller Center. " + "Enjoy the artistic and cultural experiences!" ) expected_messages = messages_to_dict(