doc: Update README.md

childmindresearch · Jan 14, 2025 · 3ae3169 · 3ae3169
1 parent 78f8819
commit 3ae3169
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -29,12 +29,10 @@ client = cloai.OpenAiLlm(api_key="your_key", model="gpt-4o")
 
 ```python
 import cloai
-import instructor
 
-client = cloai.OpenAiLlm(
-  api_key="your_key", model="llama3.2",
+client = cloai.OllamaLlm(
+  model="llama3.2",
   base_url="http://localhost:11434/v1",
-  instructor_mode=instructor.Mode.JSON
 )
 ```
 

diff --git a/src/cloai/__init__.py b/src/cloai/__init__.py
@@ -2,6 +2,13 @@
 
 from cloai.llm.bedrock import AnthropicBedrockLlm
 from cloai.llm.llm import LargeLanguageModel
+from cloai.llm.ollama import OllamaLlm
 from cloai.llm.openai import AzureLlm, OpenAiLlm
 
-__all__ = ("AnthropicBedrockLlm", "AzureLlm", "LargeLanguageModel", "OpenAiLlm")
+__all__ = (
+    "AnthropicBedrockLlm",
+    "AzureLlm",
+    "LargeLanguageModel",
+    "OllamaLlm",
+    "OpenAiLlm",
+)