seekrai 0.5.1__tar.gz → 0.5.2__tar.gz

{seekrai-0.5.1 → seekrai-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.1
+Version: 0.5.2
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow
@@ -70,7 +70,7 @@ from seekrai import SeekrFlow
 client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 
 response = client.chat.completions.create(
-    model="meta-llama-3-8b-instruct",
+    model="meta-llama/Llama-3.1-8B-Instruct",
     messages=[{"role": "user", "content": "tell me about new york"}],
 )
 print(response.choices[0].message.content)
@@ -84,7 +84,7 @@ from seekrai import SeekrFlow
 
 client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 stream = client.chat.completions.create(
-    model="meta-llama-3-8b-instruct",
+    model="meta-llama/Llama-3.1-8B-Instruct",
     messages=[{"role": "user", "content": "tell me about new york"}],
     stream=True,
 )
@@ -110,7 +110,7 @@ async def async_chat_completion(messages):
     async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
     tasks = [
         async_client.chat.completions.create(
-            model="meta-llama-3-8b-instruct",
+            model="meta-llama/Llama-3.1-8B-Instruct",
             messages=[{"role": "user", "content": message}],
         )
         for message in messages
@@ -151,7 +151,7 @@ client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 
 client.fine_tuning.create(
     training_file='file-d0d318cb-b7d9-493a-bd70-1cfe089d3815',
-    model='meta-llama-3-8b-instruct',
+    model='meta-llama/Llama-3.1-8B-Instruct',
     n_epochs=3,
     n_checkpoints=1,
     batch_size=4,

{seekrai-0.5.1 → seekrai-0.5.2}/README.md

@@ -37,7 +37,7 @@ from seekrai import SeekrFlow
 client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 
 response = client.chat.completions.create(
-    model="meta-llama-3-8b-instruct",
+    model="meta-llama/Llama-3.1-8B-Instruct",
     messages=[{"role": "user", "content": "tell me about new york"}],
 )
 print(response.choices[0].message.content)
@@ -51,7 +51,7 @@ from seekrai import SeekrFlow
 
 client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 stream = client.chat.completions.create(
-    model="meta-llama-3-8b-instruct",
+    model="meta-llama/Llama-3.1-8B-Instruct",
     messages=[{"role": "user", "content": "tell me about new york"}],
     stream=True,
 )
@@ -77,7 +77,7 @@ async def async_chat_completion(messages):
     async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
     tasks = [
         async_client.chat.completions.create(
-            model="meta-llama-3-8b-instruct",
+            model="meta-llama/Llama-3.1-8B-Instruct",
             messages=[{"role": "user", "content": message}],
         )
         for message in messages
@@ -118,7 +118,7 @@ client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
 
 client.fine_tuning.create(
     training_file='file-d0d318cb-b7d9-493a-bd70-1cfe089d3815',
-    model='meta-llama-3-8b-instruct',
+    model='meta-llama/Llama-3.1-8B-Instruct',
     n_epochs=3,
     n_checkpoints=1,
     batch_size=4,

{seekrai-0.5.1 → seekrai-0.5.2}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.1"
+version = "0.5.2"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]

seekrai-0.5.2/src/seekrai/types/agents/tools/schemas/file_search_env.py

@@ -0,0 +1,16 @@
+from pydantic import Field
+
+from seekrai.types.agents.tools.env_model_config import EnvConfig
+
+
+# TODO: figure out better way of creating tool environment models (within tool ideally), but retaining separate model_configs
+class FileSearchEnv(EnvConfig):
+    file_search_index: str = Field(min_length=1)
+    document_tool_desc: str = Field(min_length=1)
+    top_k: int = Field(
+        default=10, ge=1, le=100, description="Top K must be >= 1 and <= 100"
+    )
+    score_threshold: float = Field(
+        default=0, ge=0, lt=1.0, description="Score must be ≥ 0.0 and < 1.0"
+    )
+

seekrai-0.5.1/src/seekrai/types/agents/tools/schemas/file_search_env.py

@@ -1,11 +0,0 @@
-from pydantic import Field
-
-from seekrai.types.agents.tools.env_model_config import EnvConfig
-
-
-# TODO: figure out better way of creating tool environment models (within tool ideally), but retaining separate model_configs
-class FileSearchEnv(EnvConfig):
-    file_search_index: str
-    document_tool_desc: str
-    top_k: int = Field(default=10)
-    score_threshold: int = Field(default=0)