cleaning some examples a bit

Bam4d · Bam4d · commit b021a566cc8e · 2023-12-09T10:25:48.000Z
diff --git a/README.md b/README.md
@@ -32,8 +32,8 @@ poetry run python chat_no_streaming.py
 ### Using poetry shell
 
 ```bash
-cd examples
 poetry shell
+cd examples
 
 >> python chat_no_streaming.py
 ```
diff --git a/examples/async_chat_no_streaming.py b/examples/async_chat_no_streaming.py
@@ -15,7 +15,10 @@ async def main():
         model=model,
         messages=[ChatMessage(role="user", content="What is the best French cheese?")],
     )
-    print(chat_response)
+
+    print(chat_response.choices[0].message.content)
+
+    await client.close()
 
 
 if __name__ == "__main__":
diff --git a/examples/async_chat_with_streaming.py b/examples/async_chat_with_streaming.py
@@ -11,11 +11,17 @@ async def main():
 
     client = MistralAsyncClient(api_key=api_key)
 
+    print("Chat response:")
     async for chunk in client.chat_stream(
         model=model,
         messages=[ChatMessage(role="user", content="What is the best French cheese?")],
     ):
-        print(chunk)
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].delta.content, end="")
+
+    print("\n")
+
+    await client.close()
 
 
 if __name__ == "__main__":
diff --git a/examples/chat_no_streaming.py b/examples/chat_no_streaming.py
@@ -14,7 +14,7 @@ def main():
         model=model,
         messages=[ChatMessage(role="user", content="What is the best French cheese?")],
     )
-    print(chat_response)
+    print(chat_response.choices[0].message.content)
 
 
 if __name__ == "__main__":
diff --git a/examples/chat_with_streaming.py b/examples/chat_with_streaming.py
@@ -14,7 +14,8 @@ def main():
         model=model,
         messages=[ChatMessage(role="user", content="What is the best French cheese?")],
     ):
-        print(chunk)
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].delta.content, end="")
 
 
 if __name__ == "__main__":
diff --git a/src/mistralai/async_client.py b/src/mistralai/async_client.py
@@ -174,6 +174,9 @@ def __init__(
             timeout=timeout,
         )
 
+    async def close(self) -> None:
+        await self._backend.close()
+
     async def _request(
         self,
         method: str,
@@ -266,7 +269,6 @@ async def chat_stream(
 
         async with async_response as response:
             async for line in response.content:
-                self._logger.debug(f"Received line: {line.decode()}")
                 if line == b"\n":
                     continue
 
diff --git a/src/mistralai/client.py b/src/mistralai/client.py
@@ -1,7 +1,8 @@
+import os
 import posixpath
 from json import JSONDecodeError
 from typing import Any, Dict, Iterable, List, Optional, Union
-import os
+
 import orjson
 import requests
 from requests import Response
diff --git a/src/mistralai/client_base.py b/src/mistralai/client_base.py
@@ -35,7 +35,7 @@ def _make_chat_request(
         request_data: Dict[str, Any] = {
             "model": model,
             "messages": [msg.model_dump() for msg in messages],
-            "safe_mode": safe_mode,
+            "safe_prompt": safe_mode,
         }
         if temperature is not None:
             request_data["temperature"] = temperature
diff --git a/src/mistralai/constants.py b/src/mistralai/constants.py
@@ -2,4 +2,4 @@
 
 RETRY_STATUS_CODES = {429, 500, 502, 503, 504}
 
-ENDPOINT = "http://api.mistral.ai"
+ENDPOINT = "https://api.mistral.ai"

Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@ def main():`
`14`	`14`	`model=model,`
`15`	`15`	`messages=[ChatMessage(role="user", content="What is the best French cheese?")],`
`16`	`16`	`)`
`17`		`- print(chat_response)`
	`17`	`+ print(chat_response.choices[0].message.content)`
`18`	`18`
`19`	`19`
`20`	`20`	`if __name__ == "__main__":`
Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ def _make_chat_request(`
`35`	`35`	`request_data: Dict[str, Any] = {`
`36`	`36`	`"model": model,`
`37`	`37`	`"messages": [msg.model_dump() for msg in messages],`
`38`		`- "safe_mode": safe_mode,`
	`38`	`+ "safe_prompt": safe_mode,`
`39`	`39`	`}`
`40`	`40`	`if temperature is not None:`
`41`	`41`	`request_data["temperature"] = temperature`
Original file line number	Diff line number	Diff line change
`@@ -2,4 +2,4 @@`
`2`	`2`
`3`	`3`	`RETRY_STATUS_CODES = {429, 500, 502, 503, 504}`
`4`	`4`
`5`		`-ENDPOINT = "http://api.mistral.ai"`
	`5`	`+ENDPOINT = "https://api.mistral.ai"`