morphik 0.2.2__tar.gz → 0.2.4__tar.gz

{morphik-0.2.2 → morphik-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: morphik
-Version: 0.2.2
+Version: 0.2.4
 Summary: Morphik Python Client
 Author-email: Morphik <founders@morphik.ai>
 Requires-Python: >=3.8

{morphik-0.2.2 → morphik-0.2.4}/morphik/__init__.py

@@ -12,4 +12,4 @@ __all__ = [
     "Document",
 ]
 
-__version__ = "0.2.2"
+__version__ = "0.2.4"

{morphik-0.2.2 → morphik-0.2.4}/morphik/_internal.py

@@ -253,6 +253,7 @@ class _MorphikClientLogic:
         end_user_id: Optional[str],
         chat_id: Optional[str] = None,
         schema: Optional[Union[Type[BaseModel], Dict[str, Any]]] = None,
+        llm_config: Optional[Dict[str, Any]] = None,
     ) -> Dict[str, Any]:
         """Prepare request for query endpoint"""
         payload = {
@@ -274,6 +275,8 @@ class _MorphikClientLogic:
             payload["end_user_id"] = end_user_id
         if chat_id:
             payload["chat_id"] = chat_id
+        if llm_config:
+            payload["llm_config"] = llm_config
 
         # Add schema to payload if provided
         if schema:

{morphik-0.2.2 → morphik-0.2.4}/morphik/async_.py

@@ -356,6 +356,7 @@ class AsyncFolder:
         additional_folders: Optional[List[str]] = None,
         schema: Optional[Union[Type[BaseModel], Dict[str, Any]]] = None,
         chat_id: Optional[str] = None,
+        llm_config: Optional[Dict[str, Any]] = None,
     ) -> CompletionResponse:
         """
         Generate completion using relevant chunks as context within this folder.
@@ -395,6 +396,7 @@ class AsyncFolder:
             None,
             chat_id,
             schema,
+            llm_config,
         )
 
         # Add schema to payload if provided
@@ -892,6 +894,7 @@ class AsyncUserScope:
         additional_folders: Optional[List[str]] = None,
         schema: Optional[Union[Type[BaseModel], Dict[str, Any]]] = None,
         chat_id: Optional[str] = None,
+        llm_config: Optional[Dict[str, Any]] = None,
     ) -> CompletionResponse:
         """
         Generate completion using relevant chunks as context, scoped to the end user.
@@ -931,6 +934,7 @@ class AsyncUserScope:
             self._end_user_id,
             chat_id,
             schema,
+            llm_config,
         )
 
         # Add schema to payload if provided
@@ -1556,6 +1560,7 @@ class AsyncMorphik:
         folder_name: Optional[Union[str, List[str]]] = None,
         chat_id: Optional[str] = None,
         schema: Optional[Union[Type[BaseModel], Dict[str, Any]]] = None,
+        llm_config: Optional[Dict[str, Any]] = None,
     ) -> CompletionResponse:
         """
         Generate completion using relevant chunks as context.
@@ -1575,6 +1580,7 @@ class AsyncMorphik:
             prompt_overrides: Optional customizations for entity extraction, resolution, and query prompts
                 Either a QueryPromptOverrides object or a dictionary with the same structure
             schema: Optional schema for structured output, can be a Pydantic model or a JSON schema dict
+            llm_config: Optional LiteLLM-compatible model configuration (e.g., model name, API key, base URL)
         Returns:
             CompletionResponse
 
@@ -1662,6 +1668,7 @@ class AsyncMorphik:
             None,
             chat_id,
             schema,
+            llm_config,
         )
 
         # Add schema to payload if provided

{morphik-0.2.2 → morphik-0.2.4}/morphik/rules.py

@@ -36,7 +36,22 @@ class MetadataExtractionRule(Rule):
     def to_dict(self) -> Dict[str, Any]:
         if isinstance(self.schema, type) and issubclass(self.schema, BaseModel):
             # Convert Pydantic model to dict schema
-            schema_dict = self.schema.model_json_schema()
+            json_schema = self.schema.model_json_schema()
+            # Extract the properties from the JSON schema format
+            # to match the expected server format
+            schema_dict = {}
+            if "properties" in json_schema:
+                for field_name, field_info in json_schema["properties"].items():
+                    # Get field description from the Pydantic field
+                    description = field_info.get("description", f"No description for {field_name}")
+                    field_type = field_info.get("type", "string")
+                    schema_dict[field_name] = {
+                        "type": field_type,
+                        "description": description
+                    }
+            else:
+                # Fallback if properties not found
+                schema_dict = json_schema
         else:
             # Assume it's already a dict schema
             schema_dict = self.schema

{morphik-0.2.2 → morphik-0.2.4}/morphik/sync.py

@@ -369,6 +369,7 @@ class Folder:
         additional_folders: Optional[List[str]] = None,
         schema: Optional[Union[Type[BaseModel], Dict[str, Any]]] = None,
         chat_id: Optional[str] = None,
+        llm_config: Optional[Dict[str, Any]] = None,
     ) -> CompletionResponse:
         """
         Generate completion using relevant chunks as context within this folder.
@@ -408,6 +409,7 @@ class Folder:
             None,  # end_user_id not supported at this level
             chat_id,
             schema,
+            llm_config,
         )
 
         # Add schema to payload if provided
@@ -952,6 +954,7 @@ class UserScope:
         additional_folders: Optional[List[str]] = None,
         schema: Optional[Union[Type[BaseModel], Dict[str, Any]]] = None,
         chat_id: Optional[str] = None,
+        llm_config: Optional[Dict[str, Any]] = None,
     ) -> CompletionResponse:
         """
         Generate completion using relevant chunks as context as this end user.
@@ -991,6 +994,7 @@ class UserScope:
             self._end_user_id,
             chat_id,
             schema,
+            llm_config,
         )
 
         # Add schema to payload if provided
@@ -1699,6 +1703,7 @@ class Morphik:
         folder_name: Optional[Union[str, List[str]]] = None,
         chat_id: Optional[str] = None,
         schema: Optional[Union[Type[BaseModel], Dict[str, Any]]] = None,
+        llm_config: Optional[Dict[str, Any]] = None,
     ) -> CompletionResponse:
         """
         Generate completion using relevant chunks as context.
@@ -1719,6 +1724,7 @@ class Morphik:
                 Either a QueryPromptOverrides object or a dictionary with the same structure
             folder_name: Optional folder name to further scope operations
             schema: Optional schema for structured output, can be a Pydantic model or a JSON schema dict
+            llm_config: Optional LiteLLM-compatible model configuration (e.g., model name, API key, base URL)
         Returns:
             CompletionResponse
 
@@ -1806,6 +1812,7 @@ class Morphik:
             None,  # end_user_id not supported at this level
             chat_id,
             schema,
+            llm_config,
         )
 
         # Add schema to payload if provided

{morphik-0.2.2 → morphik-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "morphik"
-version = "0.2.2"
+version = "0.2.4"
 authors = [
     { name = "Morphik", email = "founders@morphik.ai" },
 ]