seekrai 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. seekrai/types/agents/tools/schemas/file_search_env.py +9 -4
  2. {seekrai-0.5.1.dist-info → seekrai-0.5.2.dist-info}/METADATA +5 -5
  3. {seekrai-0.5.1.dist-info → seekrai-0.5.2.dist-info}/RECORD +6 -6
  4. {seekrai-0.5.1.dist-info → seekrai-0.5.2.dist-info}/LICENSE +0 -0
  5. {seekrai-0.5.1.dist-info → seekrai-0.5.2.dist-info}/WHEEL +0 -0
  6. {seekrai-0.5.1.dist-info → seekrai-0.5.2.dist-info}/entry_points.txt +0 -0
seekrai/types/agents/tools/schemas/file_search_env.py CHANGED
@@ -5,7 +5,12 @@ from seekrai.types.agents.tools.env_model_config import EnvConfig
5
5
 
6
6
  # TODO: figure out better way of creating tool environment models (within tool ideally), but retaining separate model_configs
7
7
  class FileSearchEnv(EnvConfig):
8
- file_search_index: str
9
- document_tool_desc: str
10
- top_k: int = Field(default=10)
11
- score_threshold: int = Field(default=0)
8
+ file_search_index: str = Field(min_length=1)
9
+ document_tool_desc: str = Field(min_length=1)
10
+ top_k: int = Field(
11
+ default=10, ge=1, le=100, description="Top K must be >= 1 and <= 100"
12
+ )
13
+ score_threshold: float = Field(
14
+ default=0, ge=0, lt=1.0, description="Score must be ≥ 0.0 and < 1.0"
15
+ )
16
+
{seekrai-0.5.1.dist-info → seekrai-0.5.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: seekrai
3
- Version: 0.5.1
3
+ Version: 0.5.2
4
4
  Summary: Python client for SeekrAI
5
5
  License: Apache-2.0
6
6
  Author: SeekrFlow
@@ -70,7 +70,7 @@ from seekrai import SeekrFlow
70
70
  client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
71
71
 
72
72
  response = client.chat.completions.create(
73
- model="meta-llama-3-8b-instruct",
73
+ model="meta-llama/Llama-3.1-8B-Instruct",
74
74
  messages=[{"role": "user", "content": "tell me about new york"}],
75
75
  )
76
76
  print(response.choices[0].message.content)
@@ -84,7 +84,7 @@ from seekrai import SeekrFlow
84
84
 
85
85
  client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
86
86
  stream = client.chat.completions.create(
87
- model="meta-llama-3-8b-instruct",
87
+ model="meta-llama/Llama-3.1-8B-Instruct",
88
88
  messages=[{"role": "user", "content": "tell me about new york"}],
89
89
  stream=True,
90
90
  )
@@ -110,7 +110,7 @@ async def async_chat_completion(messages):
110
110
  async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
111
111
  tasks = [
112
112
  async_client.chat.completions.create(
113
- model="meta-llama-3-8b-instruct",
113
+ model="meta-llama/Llama-3.1-8B-Instruct",
114
114
  messages=[{"role": "user", "content": message}],
115
115
  )
116
116
  for message in messages
@@ -151,7 +151,7 @@ client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
151
151
 
152
152
  client.fine_tuning.create(
153
153
  training_file='file-d0d318cb-b7d9-493a-bd70-1cfe089d3815',
154
- model='meta-llama-3-8b-instruct',
154
+ model='meta-llama/Llama-3.1-8B-Instruct',
155
155
  n_epochs=3,
156
156
  n_checkpoints=1,
157
157
  batch_size=4,
{seekrai-0.5.1.dist-info → seekrai-0.5.2.dist-info}/RECORD RENAMED
@@ -36,7 +36,7 @@ seekrai/types/agents/tools/__init__.py,sha256=-rdY30tqMSSapMKPFOYa-6crd6tVOUvHCz
36
36
  seekrai/types/agents/tools/env_model_config.py,sha256=9POx2DPwfSXgoaziJv7QvKeMrhMsYD1exnanSRK48vw,177
37
37
  seekrai/types/agents/tools/schemas/__init__.py,sha256=sWNKRzmnWoTv3cJ5k0q0tVuWeOlRhDMEvVeClF1r4p8,202
38
38
  seekrai/types/agents/tools/schemas/file_search.py,sha256=sBH3nHgp2aIlbDsv6B3Er8LFzq6VZSXXzgN_toGVY28,289
39
- seekrai/types/agents/tools/schemas/file_search_env.py,sha256=0ZZWnaQzzYTZa-hCedmm77zn1G8JGqLpqDucK_P_Vo8,388
39
+ seekrai/types/agents/tools/schemas/file_search_env.py,sha256=T53Yd3sWoNftmxU3GnQRDBMZpzJwU0-N3vpq9LdcIXc,583
40
40
  seekrai/types/agents/tools/tool.py,sha256=VVreXuKz1LX7L1En2wTYnrCt6Za2KK2pco0RH9PxgX8,235
41
41
  seekrai/types/agents/tools/tool_env_types.py,sha256=qQDRbRBU4VGLb6uLIq_G4o-cdgZ5DBwBCa4m1Gw7OYg,151
42
42
  seekrai/types/agents/tools/tool_types.py,sha256=Auciez5JSSGjczfG38UAuuCeyRKs-ikP3bh2VAEUPFo,242
@@ -60,8 +60,8 @@ seekrai/utils/api_helpers.py,sha256=0Y8BblNIr9h_R12zdmhkxgTlxgoRkbq84QNi4nNWGu8,
60
60
  seekrai/utils/files.py,sha256=7ixn_hgV-6pEhYqLyOp-EN0o8c1CzUwJzX9n3PQ5oqo,7164
61
61
  seekrai/utils/tools.py,sha256=jgJTL-dOIouDbEJLdQpQfpXhqaz_poQYS52adyUtBjo,1781
62
62
  seekrai/version.py,sha256=q6iGQVFor8zXiPP5F-3vy9TndOxKv5JXbaNJ2kdOQws,125
63
- seekrai-0.5.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
64
- seekrai-0.5.1.dist-info/METADATA,sha256=WE8Hbfl-J_XWofa47b1BBFjItV4ny3t8AOBsG8zbwQw,4748
65
- seekrai-0.5.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
66
- seekrai-0.5.1.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
67
- seekrai-0.5.1.dist-info/RECORD,,
63
+ seekrai-0.5.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
64
+ seekrai-0.5.2.dist-info/METADATA,sha256=_uCmgAvoXVaul5oDZ-kKQ2JTHxMAoBdTvCXldbGeNXk,4780
65
+ seekrai-0.5.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
66
+ seekrai-0.5.2.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
67
+ seekrai-0.5.2.dist-info/RECORD,,