alita-sdk 0.3.212__tar.gz → 0.3.214__tar.gz

{alita_sdk-0.3.212 → alita_sdk-0.3.214}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.212
+Version: 0.3.214
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.212 → alita_sdk-0.3.214}/alita_sdk/runtime/clients/client.py

@@ -180,6 +180,7 @@ class AlitaClient:
             temperature=model_config.get("temperature"),
             max_retries=model_config.get("max_retries", 3),
             seed=model_config.get("seed", None),
+            openai_organization=str(self.project_id),
         )
          
     

{alita_sdk-0.3.212 → alita_sdk-0.3.214}/alita_sdk/runtime/toolkits/artifact.py

@@ -26,6 +26,11 @@ class ArtifactToolkit(BaseToolkit):
             connection_string = (Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                             default=None,
                                                             json_schema_extra={'secret': True})),
+            # embedding model settings
+            embedding_model=(str, Field(default="HuggingFaceEmbeddings", description="Embedding model to use")),
+            embedding_model_params=(dict, Field(default={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+                                                description="Parameters for embedding model")),
+
             __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Artifact", "icon_url": None}})
         )
     

{alita_sdk-0.3.212 → alita_sdk-0.3.214}/alita_sdk/runtime/toolkits/tools.py

@@ -76,8 +76,8 @@ def get_tools(tools_list: list, alita_client, llm, memory_store: BaseStore = Non
                 # indexer settings
                 connection_string=tool['settings'].get('connection_string', None),
                 collection_name=f"{tool.get('toolkit_name')}_{str(tool['id'])}",
-                embedding_model="HuggingFaceEmbeddings",
-                embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+                embedding_model=tool['settings'].get('embedding_model', None),
+                embedding_model_params=tool['settings'].get('embedding_model_params', None),
                 vectorstore_type="PGVector"
             ).get_tools())
         elif tool['type'] == 'vectorstore':

{alita_sdk-0.3.212 → alita_sdk-0.3.214}/alita_sdk/tools/elitea_base.py

@@ -114,7 +114,7 @@ BaseIndexDataParams = create_model(
                          description="Optional step size for progress reporting during indexing")),
     clean_index=(Optional[bool], Field(default=False,
                        description="Optional flag to enforce clean existing index before indexing new data")),
-    chunking_tool=(Literal['markdown', 'statistical', 'proposal'], Field(description="Name of chunking tool", default="markdown")),
+    chunking_tool=(Literal['','markdown', 'statistical', 'proposal'], Field(description="Name of chunking tool", default=None)),
     chunking_config=(Optional[dict], Field(description="Chunking tool configuration", default_factory=dict)),
 )
 
@@ -350,7 +350,9 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                 "alita_sdk_options": {
                     "target_schema": collection_name,
                 },
-                "connection_string": self.connection_string.get_secret_value()
+                # "connection_string": self.connection_string.get_secret_value()
+                # 'postgresql+psycopg://project_23_user:Rxu4QtM2InLVNnm62GX7@pgvector:5432/project_23'
+                "connection_string": 'postgresql+psycopg://postgres:yourpassword@localhost:5432/postgres'
             }
         elif self.vectorstore_type == 'Chroma':
             vectorstore_params = {

{alita_sdk-0.3.212 → alita_sdk-0.3.214}/alita_sdk/tools/jira/__init__.py

@@ -26,7 +26,6 @@ def get_tools(tool):
         alita=tool['settings'].get('alita', None),
         connection_string=tool['settings'].get('connection_string', None),
         collection_name=f"{tool.get('toolkit_name')}_{str(tool['id'])}",
-        doctype='code',
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector",

{alita_sdk-0.3.212 → alita_sdk-0.3.214}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.212
+Version: 0.3.214
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.212 → alita_sdk-0.3.214}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.212"
+version = "0.3.214"
 description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"