davidkhala.ai 0.1.9__tar.gz → 0.2.0__tar.gz

{davidkhala_ai-0.1.9 → davidkhala_ai-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: davidkhala.ai
-Version: 0.1.9
+Version: 0.2.0
 Summary: misc AI modules
 Requires-Python: >=3.12
 Provides-Extra: ali

davidkhala_ai-0.2.0/davidkhala/ai/agent/dify/ops/console/plugin.py

@@ -0,0 +1,68 @@
+from time import sleep
+
+from davidkhala.ai.agent.dify.ops.console import API
+from davidkhala.ai.agent.dify.ops.console.session import ConsoleUser
+
+
+class ConsolePlugin(API):
+    def __init__(self, context: ConsoleUser):
+        super().__init__()
+        self.base_url = f"{context.base_url}/workspaces/current/plugin"
+        self.session.cookies = context.session.cookies
+        self.options = context.options
+
+    def upgrade(self, *plugin_names: str) -> list[dict]:
+        versions = self.latest_version(*plugin_names)
+        self.async_install(*versions)
+
+        current = []
+        while len(current) < len(versions):
+            current = self.get(*plugin_names)
+            sleep(1)
+        return current
+
+    def async_install(self, *plugin_versioned_names: str) -> str | None:
+        url = f"{self.base_url}/install/marketplace"
+        r = self.request(url, method="POST", json={
+            'plugin_unique_identifiers': plugin_versioned_names,
+        })
+        if r['all_installed']:
+            # plugins exist, no need to install
+            return None
+
+        return r['task_id']
+
+    def plugins(self, *, page=1, size=100):
+        url = f"{self.base_url}/list?page={page}&page_size={size}"
+        r = self.request(url, method="GET")
+        _ = r['plugins']
+        assert r['total'] == len(_)
+        return _
+
+    def get(self, *plugin_names: str) -> list[dict]:
+        "inspect installed plugins"
+        url = f"{self.base_url}/list/installations/ids"
+        r = self.request(url, method="POST", json={
+            'plugin_ids': plugin_names,
+        })
+        return r['plugins']
+
+    def latest_version(self, *plugin_names: str) -> dict:
+        url = f"{self.base_url}/list/latest-versions"
+        r = self.request(url, method="POST", json={
+            'plugin_ids': plugin_names,
+        })
+        return [r['versions'][name]['unique_identifier'] for name in plugin_names]
+
+    def uninstall(self, id: str):
+        url = f"{self.base_url}/uninstall"
+
+        r = self.request(url, method="POST", json={
+            'plugin_installation_id': id
+        })
+        assert r['success'] is True
+
+    def uninstall_by(self, *plugin_names: str):
+        for name in plugin_names:
+            r = self.get(name)
+            self.uninstall(r[0]['id'])

{davidkhala_ai-0.1.9 → davidkhala_ai-0.2.0}/davidkhala/ai/agent/dify/ops/console/session.py

@@ -1,3 +1,5 @@
+from base64 import b64encode
+
 from davidkhala.ai.agent.dify.ops.console import API
 
 
@@ -11,7 +13,7 @@ class ConsoleUser(API):
 
         r = self.request(url, "POST", json={
             'email': email,
-            'password': password,
+            'password': b64encode(password.encode()).decode(), # use base64 from dify 1.11
             'remember_me': remember_me,
             'language': language,
         })

davidkhala_ai-0.2.0/davidkhala/ai/agent/dify/plugins/popular.py

@@ -0,0 +1,36 @@
+model_provider = [
+    'langgenius/siliconflow',
+    'langgenius/azure_openai',
+    'langgenius/tongyi',
+    'langgenius/jina',
+    'langgenius/openrouter',
+    'langgenius/deepseek',
+]
+
+class Knowledge:
+    data_source = [
+        'langgenius/firecrawl_datasource',
+    ]
+    chunk = [
+        'langgenius/parentchild_chunker',
+        'langgenius/general_chunker',
+    ]
+    api = [
+        'abesticode/knowledge_pro',
+    ]
+
+class Node:
+    format = [
+        'langgenius/json_process',
+        'langgenius/dify_extractor',
+    ]
+    agent = [
+        'langgenius/agent',
+    ]
+    data = [
+        'langgenius/chart',
+        'junjiem/db_query',
+        'junjiem/db_query_pre_auth',
+    ]
+
+

{davidkhala_ai-0.1.9 → davidkhala_ai-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "davidkhala.ai"
-version = "0.1.9"
+version = "0.2.0"
 description = "misc AI modules"
 readme = "README.md"
 requires-python = ">=3.12"