fix

Author: karthikscale3

src/tests/cohere/cassettes/test_cohere_chat.yaml

@@ -1,89 +1,91 @@
 interactions:
-- request:
-    body: '{"message": "Tell me a story in 3 sentences or less?", "stream": false,
-      "model": "command", "preamble": "answer like a pirate", "chat_history": [{"role":
-      "USER", "message": "Who discovered gravity?"}, {"role": "CHATBOT", "message":
-      "The man who is widely credited with discovering gravity is Sir Isaac Newton"}],
-      "connectors": [{"id": "web-search"}], "temperature": 0.1}'
-    headers:
-      accept:
-      - '*/*'
-      accept-encoding:
-      - gzip, deflate
-      connection:
-      - keep-alive
-      content-length:
-      - '370'
-      content-type:
-      - application/json
-      host:
-      - api.cohere.com
-      user-agent:
-      - python-httpx/0.27.0
-      x-fern-language:
-      - Python
-      x-fern-sdk-name:
-      - cohere
-      x-fern-sdk-version:
-      - 5.5.3
-    method: POST
-    uri: https://api.cohere.com/v1/chat
-  response:
-    body:
-      string: '{"response_id":"6b2a2c29-2cd4-4e79-a05a-db8a6019b757","text":"Argh,
-        many years ago, a bold explorer named Isaac Newton sat beneath an apple tree
-        for many hours, contemplating the nature of the universe. Suddenly, an apple
-        fell from the sky, striking him on the head. This inspired him to devise a
-        theory about the forces that govern objects on Earth and altered our understanding
-        of the universe forevermore.","generation_id":"f3b5a0d7-409b-45a7-9324-5b89c944457e","chat_history":[{"role":"USER","message":"Who
-        discovered gravity?"},{"role":"CHATBOT","message":"The man who is widely credited
-        with discovering gravity is Sir Isaac Newton"},{"role":"USER","message":"Tell
-        me a story in 3 sentences or less?"},{"role":"CHATBOT","message":"Argh, many
-        years ago, a bold explorer named Isaac Newton sat beneath an apple tree for
-        many hours, contemplating the nature of the universe. Suddenly, an apple fell
-        from the sky, striking him on the head. This inspired him to devise a theory
-        about the forces that govern objects on Earth and altered our understanding
-        of the universe forevermore."}],"finish_reason":"COMPLETE","meta":{"api_version":{"version":"1"},"billed_units":{"input_tokens":31,"output_tokens":69},"tokens":{"input_tokens":49,"output_tokens":69}},"documents":[]}'
-    headers:
-      Alt-Svc:
-      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
-      Via:
-      - 1.1 google
-      access-control-expose-headers:
-      - X-Debug-Trace-ID
-      cache-control:
-      - no-cache, no-store, no-transform, must-revalidate, private, max-age=0
-      content-length:
-      - '1270'
-      content-type:
-      - application/json
-      date:
-      - Fri, 24 May 2024 09:38:02 GMT
-      expires:
-      - Thu, 01 Jan 1970 00:00:00 UTC
-      num_chars:
-      - '220'
-      num_tokens:
-      - '100'
-      pragma:
-      - no-cache
-      server:
-      - envoy
-      vary:
-      - Origin
-      x-accel-expires:
-      - '0'
-      x-debug-trace-id:
-      - 68cf0a1a2924c2a2534e350092097d49
-      x-endpoint-monthly-call-limit:
-      - '1000'
-      x-envoy-upstream-service-time:
-      - '2573'
-      x-trial-endpoint-call-limit:
-      - '10'
-      x-trial-endpoint-call-remaining:
-      - '9'
-    status:
-      code: 200
-      message: OK
+  - request:
+      body:
+        '{"message": "Tell me a story in 3 sentences or less?", "stream": false,
+        "model": "command", "preamble": "answer like a pirate", "chat_history": [{"role":
+        "USER", "message": "Who discovered gravity?"}, {"role": "CHATBOT", "message":
+        "The man who is widely credited with discovering gravity is Sir Isaac Newton"}],
+        "connectors": [{"id": "web-search"}], "temperature": 0.1}'
+      headers:
+        accept:
+          - "*/*"
+        accept-encoding:
+          - gzip, deflate
+        connection:
+          - keep-alive
+        content-length:
+          - "370"
+        content-type:
+          - application/json
+        host:
+          - api.cohere.com
+        user-agent:
+          - python-httpx/0.27.2
+        x-fern-language:
+          - Python
+        x-fern-sdk-name:
+          - cohere
+        x-fern-sdk-version:
+          - 5.5.3
+      method: POST
+      uri: https://api.cohere.com/v1/chat
+    response:
+      body:
+        string:
+          '{"response_id":"6b2a2c29-2cd4-4e79-a05a-db8a6019b757","text":"Argh,
+          many years ago, a bold explorer named Isaac Newton sat beneath an apple tree
+          for many hours, contemplating the nature of the universe. Suddenly, an apple
+          fell from the sky, striking him on the head. This inspired him to devise a
+          theory about the forces that govern objects on Earth and altered our understanding
+          of the universe forevermore.","generation_id":"f3b5a0d7-409b-45a7-9324-5b89c944457e","chat_history":[{"role":"USER","message":"Who
+          discovered gravity?"},{"role":"CHATBOT","message":"The man who is widely credited
+          with discovering gravity is Sir Isaac Newton"},{"role":"USER","message":"Tell
+          me a story in 3 sentences or less?"},{"role":"CHATBOT","message":"Argh, many
+          years ago, a bold explorer named Isaac Newton sat beneath an apple tree for
+          many hours, contemplating the nature of the universe. Suddenly, an apple fell
+          from the sky, striking him on the head. This inspired him to devise a theory
+          about the forces that govern objects on Earth and altered our understanding
+          of the universe forevermore."}],"finish_reason":"COMPLETE","meta":{"api_version":{"version":"1"},"billed_units":{"input_tokens":31,"output_tokens":69},"tokens":{"input_tokens":49,"output_tokens":69}},"documents":[]}'
+      headers:
+        Alt-Svc:
+          - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+        Via:
+          - 1.1 google
+        access-control-expose-headers:
+          - X-Debug-Trace-ID
+        cache-control:
+          - no-cache, no-store, no-transform, must-revalidate, private, max-age=0
+        content-length:
+          - "1270"
+        content-type:
+          - application/json
+        date:
+          - Fri, 24 May 2024 09:38:02 GMT
+        expires:
+          - Thu, 01 Jan 1970 00:00:00 UTC
+        num_chars:
+          - "220"
+        num_tokens:
+          - "100"
+        pragma:
+          - no-cache
+        server:
+          - envoy
+        vary:
+          - Origin
+        x-accel-expires:
+          - "0"
+        x-debug-trace-id:
+          - 68cf0a1a2924c2a2534e350092097d49
+        x-endpoint-monthly-call-limit:
+          - "1000"
+        x-envoy-upstream-service-time:
+          - "2573"
+        x-trial-endpoint-call-limit:
+          - "10"
+        x-trial-endpoint-call-remaining:
+          - "9"
+      status:
+        code: 200
+        message: OK
 version: 1