Fix imports

maximus2600 · Mar 10, 2024 · 33c49df · 33c49df
1 parent 4a8af99
commit 33c49df
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/examples/python/bare_bones.py b/examples/python/bare_bones.py
@@ -1,4 +1,4 @@
-from mistralrs import ModelKind, MistralLoader, Request
+from mistralrs import ModelKind, MistralLoader, ChatCompletionRequest
 
 kind = ModelKind.QuantizedGGUF
 loader = MistralLoader(
@@ -11,7 +11,7 @@
 )
 runner = loader.load()
 res = runner.send_chat_completion_request(
-    Request(
+    ChatCompletionRequest(
         model="mistral",
         messages=[
             {"role": "user", "content": "Tell me a story about the Rust type system."}

diff --git a/examples/python/python_api.py b/examples/python/python_api.py
@@ -1,4 +1,4 @@
-from mistralrs import MistralLoader, QuantizedLoader, Request
+from mistralrs import MistralLoader, QuantizedLoader, ChatCompletionRequest
 
 loader = QuantizedLoader(
     MistralLoader,
@@ -11,7 +11,7 @@
 )
 runner = loader.load()
 res = runner.send_chat_completion_request(
-    Request(
+    ChatCompletionRequest(
         model="mistral",
         messages=[
             {"role": "user", "content": "Tell me a story about the Rust type system."}

diff --git a/mistralrs-pyo3/README.md b/mistralrs-pyo3/README.md
@@ -26,7 +26,7 @@ Please find [API docs here](API.md).
 
 ## Example
 ```python
-from mistralrs import ModelKind, MistralLoader, Request
+from mistralrs import ModelKind, MistralLoader, ChatCompletionRequest
 
 kind = ModelKind.QuantizedGGUF
 loader = MistralLoader(
@@ -39,7 +39,7 @@ loader = MistralLoader(
 )
 runner = loader.load()
 res = runner.send_chat_completion_request(
-    Request(
+    ChatCompletionRequest(
         model="mistral",
         messages=[
             {"role": "user", "content": "Tell me a story about the Rust type system."}