Skip to content

Commit

Permalink
Update vcr cassette
Browse files Browse the repository at this point in the history
  • Loading branch information
@pamella
pamella committed Nov 19, 2024
1 parent 0d1ae05 commit a4299e5
Show file tree
Hide file tree
Showing 2 changed files with 57 additions and 52 deletions.
93 changes: 49 additions & 44 deletions tests/test_helpers/cassettes/test_assistants/test_AIAssistant_with_rag_invoke.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand All @@ -84,7 +85,7 @@ interactions:
connection:
- keep-alive
content-length:
- '930'
- '867'
content-type:
- application/json
host:
Expand All @@ -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
Expand All @@ -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}'
Expand All @@ -149,7 +152,7 @@ interactions:
connection:
- keep-alive
content-length:
- '1105'
- '1102'
content-type:
- application/json
host:
Expand All @@ -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
Expand Down
16 changes: 8 additions & 8 deletions tests/test_helpers/test_assistants.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit a4299e5

Please sign in to comment.