kainguru-sdk 0.2.14.dev0__tar.gz → 0.2.16.dev0__tar.gz

{kainguru_sdk-0.2.14.dev0 → kainguru_sdk-0.2.16.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: kainguru-sdk
-Version: 0.2.14.dev0
+Version: 0.2.16.dev0
 Summary: Python SDK for the Kainguru ML platform — run models and fine-tune them.
 Project-URL: Homepage, https://kainguru.com
 Project-URL: Repository, https://github.com/kainguru/kainguru

{kainguru_sdk-0.2.14.dev0 → kainguru_sdk-0.2.16.dev0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "kainguru-sdk"
-version = "0.2.14.dev0"
+version = "0.2.16.dev0"
 description = "Python SDK for the Kainguru ML platform — run models and fine-tune them."
 readme = "README.md"
 requires-python = ">=3.10"

{kainguru_sdk-0.2.14.dev0 → kainguru_sdk-0.2.16.dev0}/src/kainguru/resources/_common.py

@@ -70,17 +70,19 @@ def fine_tune_args(
 
 
 def rag_ingest_args(
-    model_id: str,
     name: str,
     file_ids: list[str],
     params: dict[str, Any] | None,
-) -> tuple[dict[str, Any], dict[str, Any]]:
-    """Build the ``(query, body)`` pair for ``POST /v1/rag``."""
-    query: dict[str, Any] = {"modelId": model_id}
+) -> dict[str, Any]:
+    """Build the request body for ``POST /v1/rag``.
+
+    A knowledge base is model-independent — no model is chosen at creation (no modelId query
+    param); the model is selected per execution.
+    """
     body: dict[str, Any] = {"name": name, "fileIds": file_ids}
     if params is not None:
         body["params"] = params
-    return query, body
+    return body
 
 
 def rag_file_selection_args(file_ids: list[str]) -> dict[str, Any]:

{kainguru_sdk-0.2.14.dev0 → kainguru_sdk-0.2.16.dev0}/src/kainguru/resources/rag.py

@@ -49,7 +49,6 @@ class RagResource:
 
     def ingest(
         self,
-        model_id: str,
         name: str,
         file_ids: list[str],
         params: dict[str, Any] | None = None,
@@ -58,14 +57,13 @@ class RagResource:
     ) -> RagKnowledgeBaseDetailDto:
         """Ingest documents into a new RAG knowledge base.
 
-        Returns immediately (typically ``PENDING``).
+        A knowledge base is model-independent — no model is chosen at creation; the model is
+        selected per execution. Returns immediately (typically ``PENDING``).
         """
-        if not model_id:
-            raise KainguruConfigError("model_id is required")
         if not file_ids:
             raise KainguruConfigError("file_ids is required")
-        query, body = rag_ingest_args(model_id, name, file_ids, params)
-        response = self._t.request("POST", _PATH, params=query, json=body, api_key=api_key)
+        body = rag_ingest_args(name, file_ids, params)
+        response = self._t.request("POST", _PATH, json=body, api_key=api_key)
         return unwrap(response, RagKnowledgeBaseDetailDto)
 
     def get(self, id: str, *, api_key: str | None = None) -> RagKnowledgeBaseDetailDto:
@@ -159,20 +157,18 @@ class AsyncRagResource:
 
     async def ingest(
         self,
-        model_id: str,
         name: str,
         file_ids: list[str],
         params: dict[str, Any] | None = None,
         *,
         api_key: str | None = None,
     ) -> RagKnowledgeBaseDetailDto:
-        if not model_id:
-            raise KainguruConfigError("model_id is required")
+        # A knowledge base is model-independent — no model at creation; selected per execution.
         if not file_ids:
             raise KainguruConfigError("file_ids is required")
-        query, body = rag_ingest_args(model_id, name, file_ids, params)
+        body = rag_ingest_args(name, file_ids, params)
         response = await self._t.request(
-            "POST", _PATH, params=query, json=body, api_key=api_key
+            "POST", _PATH, json=body, api_key=api_key
         )
         return unwrap(response, RagKnowledgeBaseDetailDto)
 

{kainguru_sdk-0.2.14.dev0 → kainguru_sdk-0.2.16.dev0}/tests/test_rag.py

@@ -30,12 +30,12 @@ def _envelope(data: Any) -> dict[str, Any]:
     return {"success": True, "code": 200, "data": data}
 
 
-def test_ingest_posts_model_id_query_and_body(client: KainguruClient, respx_mock):
+def test_ingest_posts_body_no_model_id(client: KainguruClient, respx_mock):
     route = respx_mock.post(RAG_URL).mock(
         return_value=httpx.Response(200, json=_envelope(_detail()))
     )
 
-    dto = client.rag.ingest(MODEL_ID, "kb-1", [FILE_ID_1, FILE_ID_2])
+    dto = client.rag.ingest("kb-1", [FILE_ID_1, FILE_ID_2])
 
     assert str(dto.id) == KB_ID
     assert dto.status.value == "PENDING"
@@ -43,7 +43,8 @@ def test_ingest_posts_model_id_query_and_body(client: KainguruClient, respx_mock
 
     request = route.calls.last.request
     assert request.headers["X-API-Key"] == "kg_test_key"
-    assert request.url.params["modelId"] == MODEL_ID
+    # A knowledge base is model-independent: no modelId query param.
+    assert "modelId" not in request.url.params
     assert json.loads(request.content) == {
         "name": "kb-1",
         "fileIds": [FILE_ID_1, FILE_ID_2],
@@ -55,17 +56,15 @@ def test_ingest_includes_params_when_given(client: KainguruClient, respx_mock):
         return_value=httpx.Response(200, json=_envelope(_detail()))
     )
 
-    client.rag.ingest(MODEL_ID, "kb-1", [FILE_ID_1], params={"chunkSize": 512})
+    client.rag.ingest("kb-1", [FILE_ID_1], params={"chunkSize": 512})
 
     body = json.loads(route.calls.last.request.content)
     assert body == {"name": "kb-1", "fileIds": [FILE_ID_1], "params": {"chunkSize": 512}}
 
 
-def test_ingest_requires_model_id_and_file_ids(client: KainguruClient, respx_mock):
+def test_ingest_requires_file_ids(client: KainguruClient, respx_mock):
     with pytest.raises(KainguruConfigError):
-        client.rag.ingest("", "kb-1", [FILE_ID_1])
-    with pytest.raises(KainguruConfigError):
-        client.rag.ingest(MODEL_ID, "kb-1", [])
+        client.rag.ingest("kb-1", [])
 
 
 def test_get_unwraps_to_detail_dto(client: KainguruClient, respx_mock):
@@ -152,7 +151,7 @@ def test_cancel(client: KainguruClient, respx_mock):
 async def test_async_ingest(async_client: AsyncKainguruClient, respx_mock):
     respx_mock.post(RAG_URL).mock(return_value=httpx.Response(200, json=_envelope(_detail())))
 
-    dto = await async_client.rag.ingest(MODEL_ID, "kb-1", [FILE_ID_1])
+    dto = await async_client.rag.ingest("kb-1", [FILE_ID_1])
 
     assert str(dto.id) == KB_ID
     assert dto.status.value == "PENDING"