aidev-wxbot 1.0.0__tar.gz

aidev_wxbot-1.0.0/.gitignore

@@ -0,0 +1,195 @@
+### Django template
+*.log
+*.pot
+*.pyc
+__pycache__/
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+media
+
+# If your build process includes running collectstatic, then you probably don't need or want to include staticfiles/
+# in your Git repository. Update and uncomment the following line accordingly.
+# <django-project-name>/staticfiles/
+staticfiles
+
+### Python template
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+.idea/
+.codecc/
+.pdm-build
+
+pre-*-bkcodeai
+*.xml
+local_framework_settings.py
+.DS_Store
+v3logs/
+logs/
+
+### PreCI ###
+.codecc
+build.yml
+# BK CodeAI
+bkcodeai.json
+
+# apigw
+src/aidev/support-files/openapi.yaml
+src/aidev/support-files/resources.yaml
+src/aidev/support-files/apidocs/
+
+# vitepress
+src/frontend/web/docs/.vitepress/cache/
+src/frontend/web/docs/.vitepress/temp/
+src/frontend/web/docs/.vitepress/dist/
+
+# node_modules
+src/frontend/node_modules/
+src/frontend/ai-blueking/node_modules/
+src/frontend/web/node_modules/
+src/frontend/vue2-playground/node_modules/
+src/frontend/publish-template/node_modules/
+
+# lesscode dist
+src/frontend/ai-blueking/lesscode-dist/
+
+# pnpm-lock.yaml
+pnpm-lock.yaml
+
+# pnpm-workspace.yaml
+pnpm-workspace.yaml
+.aider*
+.vscode/

aidev_wxbot-1.0.0/Makefile

@@ -0,0 +1,42 @@
+# 导入子目录中的 Makefile 文件
+ROOT_DIR?=$(shell git rev-parse --show-toplevel)/src/agent
+pytest = uv run --no-sync pytest -c ${ROOT_DIR}/pyproject.toml
+
+
+.PHONY: ALL
+ALL: init
+
+uv.lock: pyproject.toml
+	uv lock
+	uv lock --check
+
+.PHONY: requirements
+requirements: requirements.txt
+
+.PHONY: requirements.txt
+requirements.txt: pyproject.toml
+	uv export --no-hashes --no-annotate --no-dev --no-editable --no-header --format requirements-txt | awk 'NR>1' >  requirements.txt
+
+.PHONY: init-project
+init: uv-install
+	@echo "Project initialization complete."
+
+.PHONY: uv-install
+uv-install:
+	uv --version
+	uv sync --inexact
+
+
+build:
+	uv pip install build
+	uv run -m build
+
+.PHONY: clean
+clean:
+	find . -name '*.pyc' -type f -delete
+	find . -name "__pycache__" | xargs rm -rf
+	find . -name ".mypy_cache"  | xargs rm -rf
+	find . -name ".cache" | xargs rm -rf
+	rm -rf ./dist
+	rm -rf .venv
+	uv clean

aidev_wxbot-1.0.0/PKG-INFO

@@ -0,0 +1,50 @@
+Metadata-Version: 2.4
+Name: aidev-wxbot
+Version: 1.0.0
+Summary: A Django app for WeChat AI bot integration with BK AI Dev platform
+Author: tencentblueking
+License: MIT
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Requires-Dist: django<5.0,>=4.0
+Requires-Dist: djangorestframework>=3.14.0
+Requires-Dist: pika<2.0,>=1.3.0
+Requires-Dist: pycryptodome>=3.23.0
+Requires-Dist: pydantic>=2.11.7
+Requires-Dist: requests>=2.32.5
+Requires-Dist: stackprinter>=0.2.12
+Description-Content-Type: text/markdown
+
+# aidev-wxbot-plugin
+
+A WeChat bot plugin for bkaidev platform.
+
+## Description
+
+This plugin provides WeChat bot functionality for the bkaidev platform, enabling automated message handling and responses.
+
+## Features
+
+- WeChat message callback handling
+- Automated message processing
+- Integration with bkaidev platform
+
+## Installation
+
+```bash
+pip install aidev_wxbot
+```
+
+## Usage
+
+Configure the plugin in your bkaidev platform and set up the WeChat bot callback URL.
+
+## License
+
+MIT License

aidev_wxbot-1.0.0/README.md

@@ -0,0 +1,27 @@
+# aidev-wxbot-plugin
+
+A WeChat bot plugin for bkaidev platform.
+
+## Description
+
+This plugin provides WeChat bot functionality for the bkaidev platform, enabling automated message handling and responses.
+
+## Features
+
+- WeChat message callback handling
+- Automated message processing
+- Integration with bkaidev platform
+
+## Installation
+
+```bash
+pip install aidev_wxbot
+```
+
+## Usage
+
+Configure the plugin in your bkaidev platform and set up the WeChat bot callback URL.
+
+## License
+
+MIT License

aidev_wxbot-1.0.0/aidev_wxbot/__init__.py

@@ -0,0 +1,12 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+aidev_wxbot Django app.
+
+A Django app for WeChat AI bot integration with BK AI Dev platform.
+"""
+
+default_app_config = "aidev_wxbot.apps.AidevWxbotConfig"
+
+__version__ = "1.0.0"
+__author__ = "tencentblueking"

aidev_wxbot-1.0.0/aidev_wxbot/api/__init__.py

@@ -0,0 +1,85 @@
+import json
+from logging import getLogger
+
+import requests
+from django.conf import settings
+
+logger = getLogger(__name__)
+
+
+class ActionFailed(Exception):
+    """
+    Action failed to execute.
+    """
+
+    def __init__(self, code, info=None):
+        self.code = code
+        self.info = info
+
+
+class Api(object):
+    def __init__(self, base_url):
+        self.base_url = base_url
+
+    def get_auth_header(self, bk_username):
+        return {}
+
+    def call_action(self, action: str, method="POST", bk_username="admin", **params):
+        if not bk_username:
+            bk_username = (
+                params.get("json", {}).get("bk_username", None)
+                or params.get("params", {}).get("bk_username", None)
+                or "admin"
+            )
+        auth_header = self.get_auth_header(bk_username)
+        url = f"{self.base_url}/{action}"
+        if "headers" not in params:
+            params["headers"] = auth_header
+        else:
+            params["headers"].update(auth_header)
+        logger.info(
+            f"\n-----call action-----\n {method}\n {url}\n"
+            f" {json.dumps(params, ensure_ascii=False)} \n-------------------"
+        )
+        try:
+            response = requests.request(method, url, **params)
+            logger.info(f"\n-----resp-----\n {response.url} {response.text}")
+            if 200 <= response.status_code < 300:
+                result = response.json()
+                if isinstance(result, dict):
+                    if (
+                        result.get("result", False)
+                        or result.get("code", -1) == 0
+                        or result.get("status", -1) != -1
+                        or result.get("errcode", 1) == 0
+                    ):
+                        return result.get("data", result)
+                    logger.error(result)
+                    raise ActionFailed(code=result.get("code"), info=result)
+
+            logger.error(response.text)
+        except Exception as e:
+            logger.error(e)
+            raise e
+
+
+class BkApi(Api):
+    BK_APIGW_MANAGER_URL_TMPL = settings.BK_APIGW_MANAGER_URL_TMPL
+    BKPAAS_ENVIRONMENT = settings.BKPAAS_ENVIRONMENT
+    BKPAAS_APP_SECRET = settings.BKPAAS_APP_SECRET
+    BKPAAS_APP_CODE = settings.BKPAAS_APP_CODE
+
+    def __init__(self, api_name):
+        base_url = f"{self.BK_APIGW_MANAGER_URL_TMPL}/{self.BKPAAS_ENVIRONMENT}".format(api_name=api_name)
+        super().__init__(base_url)
+
+    def get_auth_header(self, bk_username):
+        return {
+            "X-Bkapi-Authorization": json.dumps(
+                {
+                    "bk_app_code": self.BKPAAS_APP_CODE,
+                    "bk_app_secret": self.BKPAAS_APP_SECRET,
+                    "bk_username": bk_username,
+                }
+            )
+        }

aidev_wxbot-1.0.0/aidev_wxbot/api/bkaidev.py

@@ -0,0 +1,34 @@
+from django.conf import settings
+
+from aidev_wxbot.api import BkApi
+
+
+class BkAiDevApi:
+    def __init__(self):
+        self.api = BkApi("bkaidev")
+
+    def retrieve_agent_channel_configs(self, channel_type):
+        return self.api.call_action(
+            f"openapi/aidev/resource/v1/agent_channel/configs/?channel_type={channel_type}/", "GET"
+        )
+
+    def convert_to_rtx(self, open_id):
+        return self.api.call_action("resource/v1/qyweixin/convert_to_userid/", "POST", json={"open_id": open_id})
+
+
+class AgentBackend:
+    def __init__(self):
+        self.api = BkApi(settings.BKPAAS_BK_PLUGIN_APIGW_NAME)
+
+    def invoke(self, content):
+        data = {
+            "inputs": {
+                "command": "chat",
+                "input": content,
+                "stream": False,
+                "chat_history": [{"role": "user", "content": content}],
+                "context": [],
+            },
+            "context": {"executor": "user"},
+        }
+        return self.api.call_action("invoke/1.0.0assistant", "POST", json=data)

aidev_wxbot-1.0.0/aidev_wxbot/apps.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Django app configuration for aidev_wxbot.
+"""
+
+from django.apps import AppConfig
+
+
+class AidevWxbotConfig(AppConfig):
+    """Configuration for aidev_wxbot Django app."""
+
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "aidev_wxbot"
+    verbose_name = "AI Dev WeChat Bot"
+
+    def ready(self):
+        """Initialize the app when Django starts."""
+        # 这里可以添加应用启动时需要执行的代码

aidev_wxbot-1.0.0/aidev_wxbot/context/__init__.py

@@ -0,0 +1,49 @@
+from typing import Union
+
+from pydantic import BaseConfig, BaseModel, ConfigDict, Field
+
+from aidev_wxbot.context.message import Message
+from aidev_wxbot.utils import AttrDict
+
+
+class Context(BaseModel):
+    # config类信息：
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True, json_encoders={BaseConfig: lambda v: v.to_dict(), AttrDict: lambda v: v.to_dict()}
+    )
+    msg_proxy_config: Union[BaseConfig, AttrDict, None] = Field(default=None, description="msg proxy config")
+
+    # message相关：
+    message: Union[Message, None] = Field(default=None, description="message content")
+    # message来源信息：
+    from_type: Union[str, None] = Field(default=None, description="whether message is from group or single")
+    sender_id: Union[str, None] = Field(default=None, description="sender id")
+    sender_code: Union[str, None] = Field(default=None, description="sender code")
+    group_id: Union[str, None] = Field(default=None, description="message group id")
+    create_time: Union[str, None] = Field(default=None, description="message send time, default str")
+    to_me: bool = Field(default=False, description="whether message is send to bot")
+
+    # api：
+    payload: Union[dict, None] = Field(default=None, description="params from api request")
+    action: Union[str, None] = Field(default=None, description="api action name")
+
+    # other param：
+    param_value: Union[str, None] = Field(default=None, description="set param value in this attribute")
+    param_name: Union[str, None] = Field(default=None, description="set waiting param name in this attribute")
+
+    def __getitem__(self, item):
+        """
+        Compatible with older versions using like a dict
+        :param item:
+        :return:
+        """
+        return getattr(self, item)
+
+    def get(self, item, default=None):
+        """
+        Compatible with older versions using like a dict
+        :param default:
+        :param item:
+        :return:
+        """
+        return getattr(self, item, default)