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

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

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "kainguru-sdk"
-version = "0.2.12.dev0"
+version = "0.2.14.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.12.dev0 → kainguru_sdk-0.2.14.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
@@ -243,6 +263,13 @@ class ModelExecModelRefDto(BaseModel):
     version: str | None = None
 
 
+class RagRequestDto(BaseModel):
+    handle: str | None = None
+    topK: int | None = None
+    minScore: float | None = None
+    promptMode: str | None = None
+
+
 class SourceDto(BaseModel):
     sourceFile: str | None = None
     chunkIndex: int | None = None
@@ -253,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
 
@@ -264,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
@@ -336,6 +342,8 @@ class ExecutionResultDto(BaseModel):
     sources: list[SourceDto] | None = None
     loraRunId: str | None = None
     ragHandle: str | None = None
+    ragRequest: RagRequestDto | None = None
+    ragKnowledgeBaseId: UUID | None = None
 
 
 class ModelExecResponseDto(BaseModel):

{kainguru_sdk-0.2.12.dev0 → kainguru_sdk-0.2.14.dev0}/tests/resources/openapi.json

@@ -94,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": {
@@ -1361,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": {
@@ -1450,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": [
@@ -1574,9 +1456,6 @@
           "status": {
             "$ref": "#/components/schemas/ModelStatus"
           },
-          "model": {
-            "$ref": "#/components/schemas/ModelListDto"
-          },
           "files": {
             "type": "array",
             "items": {
@@ -1764,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": {
@@ -1948,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": {
@@ -2098,6 +2087,13 @@
           },
           "ragHandle": {
             "type": "string"
+          },
+          "ragRequest": {
+            "$ref": "#/components/schemas/RagRequestDto"
+          },
+          "ragKnowledgeBaseId": {
+            "type": "string",
+            "format": "uuid"
           }
         }
       },
@@ -2175,6 +2171,25 @@
           }
         }
       },
+      "RagRequestDto": {
+        "type": "object",
+        "properties": {
+          "handle": {
+            "type": "string"
+          },
+          "topK": {
+            "type": "integer",
+            "format": "int32"
+          },
+          "minScore": {
+            "type": "number",
+            "format": "double"
+          },
+          "promptMode": {
+            "type": "string"
+          }
+        }
+      },
       "SourceDto": {
         "type": "object",
         "properties": {
@@ -2203,9 +2218,6 @@
           "status": {
             "$ref": "#/components/schemas/ModelStatus"
           },
-          "model": {
-            "$ref": "#/components/schemas/ModelExecModelRefDto"
-          },
           "createDate": {
             "type": "string",
             "format": "date-time"