alita-sdk 0.3.205__py3-none-any.whl → 0.3.207__py3-none-any.whl

alita_sdk/tools/gitlab/__init__.py

@@ -1,16 +1,18 @@
 from typing import Dict, List, Literal, Optional
 
+from alita_sdk.tools.base.tool import BaseAction
 from langchain_core.tools import BaseTool
 from langchain_core.tools import BaseToolkit
 from pydantic import create_model, BaseModel, ConfigDict, SecretStr
 from pydantic.fields import Field
 
 from .api_wrapper import GitLabAPIWrapper
-from .tools import __all__
+# from .tools import __all__
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
 name = "gitlab"
 
+
 def get_tools(tool):
     return AlitaGitlabToolkit().get_toolkit(
         selected_tools=tool['settings'].get('selected_tools', []),
@@ -18,6 +20,15 @@ def get_tools(tool):
         repository=tool['settings']['repository'],
         branch=tool['settings']['branch'],
         private_token=tool['settings']['private_token'],
+
+        llm=tool['settings'].get('llm', None),
+        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",
         toolkit_name=tool.get('toolkit_name')
     ).get_tools()
 
@@ -27,10 +38,8 @@ class AlitaGitlabToolkit(BaseToolkit):
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
-        selected_tools = {}
-        for t in __all__:
-            default = t['tool'].__pydantic_fields__['args_schema'].default
-            selected_tools[t['name']] = default.schema() if default else default
+        selected_tools = {x['name']: x['args_schema'].schema() for x in
+                          GitLabAPIWrapper.model_construct().get_available_tools()}
         AlitaGitlabToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
@@ -66,15 +75,18 @@ class AlitaGitlabToolkit(BaseToolkit):
             selected_tools = []
         gitlab_api_wrapper = GitLabAPIWrapper(**kwargs)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
-        available_tools: List[Dict] = __all__
+        available_tools: List[Dict] = gitlab_api_wrapper.get_available_tools()
         tools = []
         for tool in available_tools:
             if selected_tools:
-                if tool['name'] not in selected_tools:
+                if tool["name"] not in selected_tools:
                     continue
-                initiated_tool = tool['tool'](api_wrapper=gitlab_api_wrapper)
-                initiated_tool.name = prefix + tool['name']
-                tools.append(initiated_tool)
+            tools.append(BaseAction(
+                api_wrapper=gitlab_api_wrapper,
+                name=prefix + tool["name"],
+                description=tool["description"] + "\nrepo: " + gitlab_api_wrapper.repository,
+                args_schema=tool["args_schema"]
+            ))
         return cls(tools=tools)
 
     def get_tools(self):