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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: kainguru-sdk
-Version: 0.2.13.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.13.dev0 → kainguru_sdk-0.2.16.dev0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "kainguru-sdk"
-version = "0.2.13.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.13.dev0 → kainguru_sdk-0.2.16.dev0}/src/kainguru/models.py

@@ -16,15 +16,6 @@ class RagIngestRequestDto(BaseModel):
     params: dict[str, Any] | None = None
 
 
-class CategoryShortDto(BaseModel):
-    id: UUID | None = None
-    slug: str | None = None
-    name: str | None = None
-    icon: str | None = None
-    createDate: AwareDatetime | None = None
-    modifiedDate: AwareDatetime | None = None
-
-
 class FileType(Enum):
     JPEG = 'JPEG'
     PNG = 'PNG'
@@ -75,27 +66,6 @@ class FileDto(BaseModel):
     accessLevel: AccessLevel | None = None
 
 
-class AccessType(Enum):
-    PUBLIC = 'PUBLIC'
-    PRIVATE = 'PRIVATE'
-
-
-class GpuType(Enum):
-    B200 = 'B200'
-    H200 = 'H200'
-    H100 = 'H100'
-    A100 = 'A100'
-    V100 = 'V100'
-    MI300X = 'MI300X'
-    L40S = 'L40S'
-    RTX_5090 = 'RTX_5090'
-    RTX_4090 = 'RTX_4090'
-    L4 = 'L4'
-    RTX_4070_SUPER = 'RTX_4070_SUPER'
-    RTX_4060_TI = 'RTX_4060_TI'
-    RX_7900_XTX = 'RX_7900_XTX'
-
-
 class ModelStatus(Enum):
     PENDING = 'PENDING'
     RUNNING = 'RUNNING'
@@ -150,6 +120,15 @@ class ModelFineTuneRequestDto(BaseModel):
     input: dict[str, Any]
 
 
+class CategoryShortDto(BaseModel):
+    id: UUID | None = None
+    slug: str | None = None
+    name: str | None = None
+    icon: str | None = None
+    createDate: AwareDatetime | None = None
+    modifiedDate: AwareDatetime | None = None
+
+
 class FineTuneResultDto(BaseModel):
     id: UUID | None = None
     status: ModelStatus | None = None
@@ -198,6 +177,47 @@ class ModelExecInputDto(BaseModel):
     requiredWhen: dict[str, Any] | None = None
 
 
+class AccessType(Enum):
+    PUBLIC = 'PUBLIC'
+    PRIVATE = 'PRIVATE'
+
+
+class GpuType(Enum):
+    B200 = 'B200'
+    H200 = 'H200'
+    H100 = 'H100'
+    A100 = 'A100'
+    V100 = 'V100'
+    MI300X = 'MI300X'
+    L40S = 'L40S'
+    RTX_5090 = 'RTX_5090'
+    RTX_4090 = 'RTX_4090'
+    L4 = 'L4'
+    RTX_4070_SUPER = 'RTX_4070_SUPER'
+    RTX_4060_TI = 'RTX_4060_TI'
+    RX_7900_XTX = 'RX_7900_XTX'
+
+
+class ModelListDto(BaseModel):
+    id: UUID | None = None
+    name: str | None = None
+    description: str | None = None
+    accessType: AccessType | None = None
+    fineTuneEnabled: bool | None = None
+    ragEnabled: bool | None = None
+    status: ModelStatus | None = None
+    version: str | None = None
+    tags: list[str] | None = None
+    executionCount: int | None = None
+    fineTuneCount: int | None = None
+    gpuType: GpuType | None = None
+    gpuMemoryGb: int | None = None
+    category: CategoryShortDto | None = None
+    createDate: AwareDatetime | None = None
+    modifiedDate: AwareDatetime | None = None
+    banner: str | None = None
+
+
 class UserDto(BaseModel):
     id: UUID | None = None
     username: str | None = None
@@ -260,7 +280,6 @@ class RagKnowledgeBaseDto(BaseModel):
     name: str | None = None
     ragHandle: str | None = None
     status: ModelStatus | None = None
-    model: ModelExecModelRefDto | None = None
     createDate: AwareDatetime | None = None
     modifiedDate: AwareDatetime | None = None
 
@@ -271,31 +290,11 @@ class RagKnowledgeBaseListApiResponse(BaseModel):
     data: list[RagKnowledgeBaseDto] | None = None
 
 
-class ModelListDto(BaseModel):
-    id: UUID | None = None
-    name: str | None = None
-    description: str | None = None
-    accessType: AccessType | None = None
-    fineTuneEnabled: bool | None = None
-    status: ModelStatus | None = None
-    version: str | None = None
-    tags: list[str] | None = None
-    executionCount: int | None = None
-    fineTuneCount: int | None = None
-    gpuType: GpuType | None = None
-    gpuMemoryGb: int | None = None
-    category: CategoryShortDto | None = None
-    createDate: AwareDatetime | None = None
-    modifiedDate: AwareDatetime | None = None
-    banner: str | None = None
-
-
 class RagKnowledgeBaseDetailDto(BaseModel):
     id: UUID | None = None
     name: str | None = None
     ragHandle: str | None = None
     status: ModelStatus | None = None
-    model: ModelListDto | None = None
     files: list[FileDto] | None = None
     latestOperation: RagOperationResultDto | None = None
     createDate: AwareDatetime | None = None
@@ -344,6 +343,7 @@ class ExecutionResultDto(BaseModel):
     loraRunId: str | None = None
     ragHandle: str | None = None
     ragRequest: RagRequestDto | None = None
+    ragKnowledgeBaseId: UUID | None = None
 
 
 class ModelExecResponseDto(BaseModel):

{kainguru_sdk-0.2.13.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.13.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.13.dev0 → kainguru_sdk-0.2.16.dev0}/tests/resources/openapi.json

@@ -35,15 +35,6 @@
             "schema": {
               "$ref": "#/components/schemas/ModelStatus"
             }
-          },
-          {
-            "name": "modelId",
-            "in": "query",
-            "required": false,
-            "schema": {
-              "type": "string",
-              "format": "uuid"
-            }
           }
         ],
         "responses": {
@@ -103,17 +94,6 @@
         ],
         "description": "Ingest documents into a new RAG knowledge base",
         "operationId": "ingestRag",
-        "parameters": [
-          {
-            "name": "modelId",
-            "in": "query",
-            "required": true,
-            "schema": {
-              "type": "string",
-              "format": "uuid"
-            }
-          }
-        ],
         "requestBody": {
           "content": {
             "application/json": {
@@ -1370,32 +1350,6 @@
           "name"
         ]
       },
-      "CategoryShortDto": {
-        "type": "object",
-        "properties": {
-          "id": {
-            "type": "string",
-            "format": "uuid"
-          },
-          "slug": {
-            "type": "string"
-          },
-          "name": {
-            "type": "string"
-          },
-          "icon": {
-            "type": "string"
-          },
-          "createDate": {
-            "type": "string",
-            "format": "date-time"
-          },
-          "modifiedDate": {
-            "type": "string",
-            "format": "date-time"
-          }
-        }
-      },
       "FileDto": {
         "type": "object",
         "properties": {
@@ -1459,87 +1413,6 @@
           }
         }
       },
-      "ModelListDto": {
-        "type": "object",
-        "properties": {
-          "id": {
-            "type": "string",
-            "format": "uuid"
-          },
-          "name": {
-            "type": "string"
-          },
-          "description": {
-            "type": "string"
-          },
-          "accessType": {
-            "type": "string",
-            "enum": [
-              "PUBLIC",
-              "PRIVATE"
-            ]
-          },
-          "fineTuneEnabled": {
-            "type": "boolean"
-          },
-          "status": {
-            "$ref": "#/components/schemas/ModelStatus"
-          },
-          "version": {
-            "type": "string"
-          },
-          "tags": {
-            "type": "array",
-            "items": {
-              "type": "string"
-            }
-          },
-          "executionCount": {
-            "type": "integer",
-            "format": "int64"
-          },
-          "fineTuneCount": {
-            "type": "integer",
-            "format": "int64"
-          },
-          "gpuType": {
-            "type": "string",
-            "enum": [
-              "B200",
-              "H200",
-              "H100",
-              "A100",
-              "V100",
-              "MI300X",
-              "L40S",
-              "RTX_5090",
-              "RTX_4090",
-              "L4",
-              "RTX_4070_SUPER",
-              "RTX_4060_TI",
-              "RX_7900_XTX"
-            ]
-          },
-          "gpuMemoryGb": {
-            "type": "integer",
-            "format": "int32"
-          },
-          "category": {
-            "$ref": "#/components/schemas/CategoryShortDto"
-          },
-          "createDate": {
-            "type": "string",
-            "format": "date-time"
-          },
-          "modifiedDate": {
-            "type": "string",
-            "format": "date-time"
-          },
-          "banner": {
-            "type": "string"
-          }
-        }
-      },
       "ModelStatus": {
         "type": "string",
         "enum": [
@@ -1583,9 +1456,6 @@
           "status": {
             "$ref": "#/components/schemas/ModelStatus"
           },
-          "model": {
-            "$ref": "#/components/schemas/ModelListDto"
-          },
           "files": {
             "type": "array",
             "items": {
@@ -1773,6 +1643,32 @@
           "name"
         ]
       },
+      "CategoryShortDto": {
+        "type": "object",
+        "properties": {
+          "id": {
+            "type": "string",
+            "format": "uuid"
+          },
+          "slug": {
+            "type": "string"
+          },
+          "name": {
+            "type": "string"
+          },
+          "icon": {
+            "type": "string"
+          },
+          "createDate": {
+            "type": "string",
+            "format": "date-time"
+          },
+          "modifiedDate": {
+            "type": "string",
+            "format": "date-time"
+          }
+        }
+      },
       "FineTuneResultDto": {
         "type": "object",
         "properties": {
@@ -1957,6 +1853,90 @@
           }
         }
       },
+      "ModelListDto": {
+        "type": "object",
+        "properties": {
+          "id": {
+            "type": "string",
+            "format": "uuid"
+          },
+          "name": {
+            "type": "string"
+          },
+          "description": {
+            "type": "string"
+          },
+          "accessType": {
+            "type": "string",
+            "enum": [
+              "PUBLIC",
+              "PRIVATE"
+            ]
+          },
+          "fineTuneEnabled": {
+            "type": "boolean"
+          },
+          "ragEnabled": {
+            "type": "boolean"
+          },
+          "status": {
+            "$ref": "#/components/schemas/ModelStatus"
+          },
+          "version": {
+            "type": "string"
+          },
+          "tags": {
+            "type": "array",
+            "items": {
+              "type": "string"
+            }
+          },
+          "executionCount": {
+            "type": "integer",
+            "format": "int64"
+          },
+          "fineTuneCount": {
+            "type": "integer",
+            "format": "int64"
+          },
+          "gpuType": {
+            "type": "string",
+            "enum": [
+              "B200",
+              "H200",
+              "H100",
+              "A100",
+              "V100",
+              "MI300X",
+              "L40S",
+              "RTX_5090",
+              "RTX_4090",
+              "L4",
+              "RTX_4070_SUPER",
+              "RTX_4060_TI",
+              "RX_7900_XTX"
+            ]
+          },
+          "gpuMemoryGb": {
+            "type": "integer",
+            "format": "int32"
+          },
+          "category": {
+            "$ref": "#/components/schemas/CategoryShortDto"
+          },
+          "createDate": {
+            "type": "string",
+            "format": "date-time"
+          },
+          "modifiedDate": {
+            "type": "string",
+            "format": "date-time"
+          },
+          "banner": {
+            "type": "string"
+          }
+        }
+      },
       "UserDto": {
         "type": "object",
         "properties": {
@@ -2110,6 +2090,10 @@
           },
           "ragRequest": {
             "$ref": "#/components/schemas/RagRequestDto"
+          },
+          "ragKnowledgeBaseId": {
+            "type": "string",
+            "format": "uuid"
           }
         }
       },
@@ -2234,9 +2218,6 @@
           "status": {
             "$ref": "#/components/schemas/ModelStatus"
           },
-          "model": {
-            "$ref": "#/components/schemas/ModelExecModelRefDto"
-          },
           "createDate": {
             "type": "string",
             "format": "date-time"

{kainguru_sdk-0.2.13.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"