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

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

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

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

@@ -243,6 +243,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
@@ -336,6 +343,7 @@ class ExecutionResultDto(BaseModel):
     sources: list[SourceDto] | None = None
     loraRunId: str | None = None
     ragHandle: str | None = None
+    ragRequest: RagRequestDto | None = None
 
 
 class ModelExecResponseDto(BaseModel):

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

@@ -35,6 +35,15 @@
             "schema": {
               "$ref": "#/components/schemas/ModelStatus"
             }
+          },
+          {
+            "name": "modelId",
+            "in": "query",
+            "required": false,
+            "schema": {
+              "type": "string",
+              "format": "uuid"
+            }
           }
         ],
         "responses": {
@@ -2098,6 +2107,9 @@
           },
           "ragHandle": {
             "type": "string"
+          },
+          "ragRequest": {
+            "$ref": "#/components/schemas/RagRequestDto"
           }
         }
       },
@@ -2175,6 +2187,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": {