cohere tests

tests/cassettes/test_cohere/test_cohere_async_chat.yaml

@@ -0,0 +1,74 @@
+interactions:
+- request:
+    body: '{"message": "Hello!", "stream": false, "max_tokens": 100}'
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '57'
+      content-type:
+      - application/json
+      host:
+      - api.cohere.ai
+      user-agent:
+      - python-httpx/0.25.2
+      x-fern-language:
+      - Python
+      x-fern-sdk-name:
+      - cohere
+      x-fern-sdk-version:
+      - 5.2.5
+    method: POST
+    uri: https://api.cohere.ai/v1/chat
+  response:
+    body:
+      string: '{"response_id":"85411c88-dcf9-4ed8-87f9-d32b2c06edfa","text":"Hello!
+        How can I help you today?","generation_id":"ea42ed56-669a-4e53-abce-1f2a6ded3c81","chat_history":[{"role":"USER","message":"Hello!"},{"role":"CHATBOT","message":"Hello!
+        How can I help you today?"}],"finish_reason":"COMPLETE","meta":{"api_version":{"version":"1"},"billed_units":{"input_tokens":2,"output_tokens":9},"tokens":{"input_tokens":68,"output_tokens":9}}}'
+    headers:
+      Alt-Svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      Content-Length:
+      - '432'
+      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-type:
+      - application/json
+      date:
+      - Fri, 12 Apr 2024 18:44:02 GMT
+      expires:
+      - Thu, 01 Jan 1970 00:00:00 UTC
+      num_chars:
+      - '421'
+      num_tokens:
+      - '11'
+      pragma:
+      - no-cache
+      server:
+      - envoy
+      vary:
+      - Origin
+      x-accel-expires:
+      - '0'
+      x-debug-trace-id:
+      - d86f0708c168a539d7010d395ba626b6
+      x-endpoint-monthly-call-limit:
+      - '1000'
+      x-envoy-upstream-service-time:
+      - '388'
+      x-trial-endpoint-call-limit:
+      - '20'
+      x-trial-endpoint-call-remaining:
+      - '18'
+    status:
+      code: 200
+      message: OK
+version: 1

tests/cassettes/test_cohere/test_cohere_async_stream_chat.yaml

@@ -0,0 +1,282 @@
+interactions:
+- request:
+    body: '{"message": "Tell me a short story", "stream": true, "max_tokens": 100}'
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '71'
+      content-type:
+      - application/json
+      host:
+      - api.cohere.ai
+      user-agent:
+      - python-httpx/0.25.2
+      x-fern-language:
+      - Python
+      x-fern-sdk-name:
+      - cohere
+      x-fern-sdk-version:
+      - 5.2.5
+    method: POST
+    uri: https://api.cohere.ai/v1/chat
+  response:
+    body:
+      string: '{"is_finished":false,"event_type":"stream-start","generation_id":"561ff9f4-1d4a-42e3-9992-ffa081b05e2a"}
+
+        {"is_finished":false,"event_type":"text-generation","text":"Once"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" upon"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" a"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" time"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" in"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" a"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" small"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" peaceful"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" town"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" nestled"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" among"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" rolling"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" hills"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" there"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" lived"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" a"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" curious"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" little"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" girl"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" named"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" Emily"}
+
+        {"is_finished":false,"event_type":"text-generation","text":"."}
+
+        {"is_finished":false,"event_type":"text-generation","text":" Emily"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" had"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" a"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" vivid"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" imagination"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" and"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" often"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" spent"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" her"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" days"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" dreaming"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" up"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" grand"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" adventures"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" and"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" far"}
+
+        {"is_finished":false,"event_type":"text-generation","text":"away"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" lands"}
+
+        {"is_finished":false,"event_type":"text-generation","text":"."}
+
+        {"is_finished":false,"event_type":"text-generation","text":"\n\nOne"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" sunny"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" afternoon"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" while"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" wandering"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" through"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" the"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" woods"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" near"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" her"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" home"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" Emily"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" stumbled"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" upon"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" an"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" old"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" forgotten"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" path"}
+
+        {"is_finished":false,"event_type":"text-generation","text":"."}
+
+        {"is_finished":false,"event_type":"text-generation","text":" Intr"}
+
+        {"is_finished":false,"event_type":"text-generation","text":"igued"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" she"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" decided"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" to"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" follow"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" it"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" eager"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" to"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" see"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" where"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" it"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" led"}
+
+        {"is_finished":false,"event_type":"text-generation","text":"."}
+
+        {"is_finished":false,"event_type":"text-generation","text":" As"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" she"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" ventured"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" deeper"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" into"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" the"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" forest"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":false,"event_type":"text-generation","text":" the"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" path"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" became"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" more"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" and"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" more"}
+
+        {"is_finished":false,"event_type":"text-generation","text":" overgrown"}
+
+        {"is_finished":false,"event_type":"text-generation","text":","}
+
+        {"is_finished":true,"event_type":"stream-end","response":{"response_id":"42bac2d4-467e-4fd3-92b4-da844006b6a3","text":"Once
+        upon a time, in a small, peaceful town nestled among rolling hills, there
+        lived a curious little girl named Emily. Emily had a vivid imagination and
+        often spent her days dreaming up grand adventures and faraway lands.\n\nOne
+        sunny afternoon, while wandering through the woods near her home, Emily stumbled
+        upon an old, forgotten path. Intrigued, she decided to follow it, eager to
+        see where it led. As she ventured deeper into the forest, the path became
+        more and more overgrown,","generation_id":"561ff9f4-1d4a-42e3-9992-ffa081b05e2a","chat_history":[{"role":"USER","message":"Tell
+        me a short story"},{"role":"CHATBOT","message":"Once upon a time, in a small,
+        peaceful town nestled among rolling hills, there lived a curious little girl
+        named Emily. Emily had a vivid imagination and often spent her days dreaming
+        up grand adventures and faraway lands.\n\nOne sunny afternoon, while wandering
+        through the woods near her home, Emily stumbled upon an old, forgotten path.
+        Intrigued, she decided to follow it, eager to see where it led. As she ventured
+        deeper into the forest, the path became more and more overgrown,"}],"finish_reason":"MAX_TOKENS","meta":{"api_version":{"version":"1"},"billed_units":{"input_tokens":5,"output_tokens":100},"tokens":{"input_tokens":71,"output_tokens":100}}},"finish_reason":"MAX_TOKENS"}
+
+        '
+    headers:
+      Alt-Svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      Transfer-Encoding:
+      - chunked
+      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-type:
+      - application/stream+json
+      date:
+      - Fri, 12 Apr 2024 18:44:06 GMT
+      expires:
+      - Thu, 01 Jan 1970 00:00:00 UTC
+      pragma:
+      - no-cache
+      server:
+      - envoy
+      vary:
+      - Origin
+      x-accel-expires:
+      - '0'
+      x-debug-trace-id:
+      - b5aa09210dbaf63f33768a582fb015ee
+      x-endpoint-monthly-call-limit:
+      - '1000'
+      x-envoy-upstream-service-time:
+      - '40'
+      x-trial-endpoint-call-limit:
+      - '20'
+      x-trial-endpoint-call-remaining:
+      - '16'
+    status:
+      code: 200
+      message: OK
+version: 1