agentUniverse 0.0.12.dev1__tar.gz → 0.0.12.dev2__tar.gz

{agentuniverse-0.0.12.dev1 → agentuniverse-0.0.12.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agentUniverse
-Version: 0.0.12.dev1
+Version: 0.0.12.dev2
 Summary: agentUniverse is a framework for developing applications powered by multi-agent base on large language model.
 Home-page: https://github.com/alipay/agentUniverse
 Author: AntGroup

agentuniverse-0.0.12.dev2/agentuniverse_product/__init__.py

@@ -0,0 +1,7 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/23 17:48
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: __init__.py.py

agentuniverse-0.0.12.dev2/agentuniverse_product/agentuniverse_product.py

@@ -0,0 +1,116 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/23 17:49
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: agentuniverse_product.py
+import sys
+
+from agentuniverse.base.agentuniverse import AgentUniverse
+from agentuniverse.base.component.application_component_manager import ApplicationComponentManager
+from agentuniverse.base.component.component_configer_util import ComponentConfigerUtil
+from agentuniverse.base.component.component_enum import ComponentEnum
+from agentuniverse.base.config.application_configer.app_configer import AppConfiger
+from agentuniverse.base.config.application_configer.application_config_manager import ApplicationConfigManager
+from agentuniverse.base.config.component_configer.component_configer import ComponentConfiger
+from agentuniverse.base.config.config_type_enum import ConfigTypeEnum
+from agentuniverse.base.config.configer import Configer
+from agentuniverse.base.util.system_util import get_project_root_path
+from agentuniverse.base.annotation.singleton import singleton
+from agentuniverse.database.sqldb_wrapper_manager import SQLDBWrapperManager
+from agentuniverse_product.base.product import Product
+from agentuniverse_product.base.product_configer import ProductConfiger
+from agentuniverse_product.base.product_manager import ProductManager
+from agentuniverse_product.dal.message_library import MESSAGE_TABLE_NAME, MessageORM
+from agentuniverse_product.dal.session_library import SESSION_TABLE_NAME, SessionORM
+
+
+@singleton
+class AgentUniverseProduct(object):
+    """Initialize the agentUniverse product."""
+
+    def __init__(self):
+        self.__application_container = ApplicationComponentManager()
+        self.__config_container: ApplicationConfigManager = ApplicationConfigManager()
+        self.__system_default_product_package = []
+
+    def start(self, config_path: str = None):
+        """Start the agentUniverse product."""
+        # get default config path
+        project_root_path = get_project_root_path()
+        sys.path.append(str(project_root_path.parent))
+        app_path = project_root_path / 'app'
+        if app_path.exists():
+            sys.path.append(str(app_path))
+        if not config_path:
+            config_path = project_root_path / 'config' / 'config.toml'
+            config_path = str(config_path)
+
+        # load the configuration file
+        configer = Configer(path=config_path).load()
+        app_configer = AppConfiger().load_by_configer(configer)
+        self.__config_container.app_configer = app_configer
+
+        # init product tables (session and message)
+        self.__init_product_tables()
+
+        # scan and register the product components
+        self.__scan_and_register_product(self.__config_container.app_configer)
+
+        # start the product ui server
+        try:
+            from magent_ui import launch
+            launch()
+        except ImportError as e:
+            raise ImportError(
+                "Could not start product server provided by magent-ui."
+                " Please install it with `pip install magent-ui ruamel.yaml`."
+            )
+
+    def __init_product_tables(self):
+        """Initialize the product tables including session and message tables."""
+        system_sqldb_wrapper = SQLDBWrapperManager().get_instance_obj('__system_db__')
+        if system_sqldb_wrapper is None:
+            raise Exception('system db has not been initialized.')
+        with system_sqldb_wrapper.sql_database._engine.connect() as conn:
+            # init session db
+            if not conn.dialect.has_table(conn, SESSION_TABLE_NAME):
+                SessionORM.metadata.create_all(
+                    system_sqldb_wrapper.sql_database._engine)
+            # init message db
+            if not conn.dialect.has_table(conn, MESSAGE_TABLE_NAME):
+                MessageORM.metadata.create_all(
+                    system_sqldb_wrapper.sql_database._engine)
+
+    def __scan_and_register_product(self, app_configer: AppConfiger):
+        """Scan the product component directory and register the product components.
+
+        Args:
+            app_configer(AppConfiger): the AppConfiger object
+        """
+        core_product_package_list = ((app_configer.core_product_package_list or app_configer.core_default_package_list)
+                                     + self.__system_default_product_package)
+        if core_product_package_list is None:
+            return
+        product_configer_list = AgentUniverse().scan(core_product_package_list, ConfigTypeEnum.YAML,
+                                                     ComponentEnum.PRODUCT)
+        self.__register(product_configer_list)
+
+    def __register(self, product_configer_list: list[ComponentConfiger]):
+        """Register the product components.
+
+        Args:
+            product_configer_list(list): the product component configer list
+        """
+        for product_configer in product_configer_list:
+            configer_instance: ProductConfiger = ProductConfiger(
+            ).load_by_configer(product_configer.configer)
+            component_clz = ComponentConfigerUtil.get_component_object_clz_by_component_configer(
+                configer_instance)
+            product_instance: Product = component_clz(
+            ).initialize_by_component_configer(configer_instance)
+            if product_instance is None:
+                continue
+            product_instance.component_config_path = product_configer.configer.path
+            ProductManager().register(product_instance.get_instance_code(), product_instance)

agentuniverse-0.0.12.dev2/agentuniverse_product/base/__init__.py

@@ -0,0 +1,7 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/23 17:48
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: __init__.py.py

agentuniverse-0.0.12.dev2/agentuniverse_product/base/agent_product.py

@@ -0,0 +1,43 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/25 22:50
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: agent_product.py
+from typing import Optional
+
+from agentuniverse.agent.agent import Agent
+from agentuniverse.base.config.application_configer.application_config_manager import ApplicationConfigManager
+from agentuniverse_product.base.product import Product
+from agentuniverse_product.base.product_configer import ProductConfiger
+
+
+class AgentProduct(Product):
+    """The basic class of the agent product."""
+
+    opening_speech: Optional[str] = None
+    _instance: Optional[Agent] = None
+
+    @property
+    def instance(self) -> Agent:
+        return self._instance
+
+    def get_instance_code(self) -> str:
+        """Return the full name of the product."""
+        appname = ApplicationConfigManager().app_configer.base_info_appname
+        return f"{appname}.product.{self.id}"
+
+    def initialize_by_component_configer(self,
+                                         product_configer: ProductConfiger) -> 'Product':
+        """Initialize the Product by the ProductConfiger object.
+
+        Args:
+            product_configer(ProductConfiger): A configer contains product basic info.
+        Returns:
+            Product: A Product instance.
+        """
+        super().initialize_by_component_configer(product_configer)
+        if hasattr(product_configer, 'opening_speech') and product_configer.opening_speech:
+            self.opening_speech = product_configer.opening_speech
+        return self

agentuniverse-0.0.12.dev2/agentuniverse_product/base/constant/__init__.py

@@ -0,0 +1,7 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/26 10:07
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: __init__.py.py

agentuniverse-0.0.12.dev2/agentuniverse_product/base/constant/llm_constant.py

@@ -0,0 +1,29 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/26 10:10
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: llm.py
+# llm model name list
+LLM_MODEL_NAME = {
+    'demo_llm': ['gpt-3.5-turbo', 'gpt-4o', 'gpt-4-turbo', 'gpt-4o-mini', 'gpt-4'],
+    'default_openai_llm': ['gpt-3.5-turbo', 'gpt-4o', 'gpt-4-turbo', 'gpt-4o-mini', 'gpt-4'],
+    'qwen_llm': ['qwen-max', 'qwen-long', 'qwen-turbo', 'qwen2-72b-instruct', 'qwen2-7b-instruct', 'qwen1.5-110b-chat',
+                 'qwen1.5-72b-chat', 'qwen1.5-7b-chat'],
+    'default_qwen_llm': ['qwen-max', 'qwen-long', 'qwen-turbo', 'qwen2-72b-instruct', 'qwen2-7b-instruct',
+                         'qwen1.5-110b-chat', 'qwen1.5-72b-chat', 'qwen1.5-7b-chat'],
+    'wenxin_llm': ['ERNIE-Speed-AppBuilder-8K-0516', 'ERNIE-Lite-8K-0725', 'ERNIE-Speed-128K', 'ERNIE-3.5-128K',
+                   'ERNIE-3.5-8K-0701', 'ERNIE-4.0-8K-0613', 'ERNIE-4.0-8K-Preview', 'ERNIE-3.5-8K-Preview',
+                   'ERNIE-Tiny-8K', 'ERNIE-4.0-8K-Latest', 'ERNIE-4.0-Turbo-8K'],
+    'default_wenxin_llm': ['ERNIE-Speed-AppBuilder-8K-0516', 'ERNIE-Lite-8K-0725', 'ERNIE-Speed-128K', 'ERNIE-3.5-128K',
+                           'ERNIE-3.5-8K-0701', 'ERNIE-4.0-8K-0613', 'ERNIE-4.0-8K-Preview', 'ERNIE-3.5-8K-Preview',
+                           'ERNIE-Tiny-8K', 'ERNIE-4.0-8K-Latest', 'ERNIE-4.0-Turbo-8K'],
+    'kimi_llm': ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'],
+    'default_kimi_llm': ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'],
+    'deep_seek_llm': ['deepseek-chat', 'deepseek-coder'],
+    'default_deepseek_llm': ['deepseek-chat', 'deepseek-coder'],
+    'baichuan_llm': ['Baichuan4', 'Baichuan3-Turbo', 'Baichuan3-Turbo-128k', 'Baichuan2-Turbo', 'Baichuan2-Turbo-192k'],
+    'default_baichuan_llm': ['Baichuan4', 'Baichuan3-Turbo', 'Baichuan3-Turbo-128k', 'Baichuan2-Turbo',
+                             'Baichuan2-Turbo-192k'],
+}

agentuniverse-0.0.12.dev2/agentuniverse_product/base/planner_product.py

@@ -0,0 +1,37 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/8/12 17:57
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: planner_product.py
+from typing import Optional, List
+
+from agentuniverse.agent.plan.planner.planner import Planner
+from agentuniverse_product.base.product import Product
+from agentuniverse_product.base.product_configer import ProductConfiger
+
+
+class PlannerProduct(Product):
+    """The basic class of the planner product."""
+
+    member_keys: Optional[List[str]] = None
+    _instance: Optional[Planner] = None
+
+    @property
+    def instance(self) -> Planner:
+        return self._instance
+
+    def initialize_by_component_configer(self,
+                                         product_configer: ProductConfiger) -> 'Product':
+        """Initialize the Product by the ProductConfiger object.
+
+        Args:
+            product_configer(ProductConfiger): A configer contains product basic info.
+        Returns:
+            Product: A Product instance.
+        """
+        super().initialize_by_component_configer(product_configer)
+        if hasattr(product_configer, 'member_keys') and product_configer.member_keys:
+            self.member_keys = product_configer.member_keys
+        return self

agentuniverse-0.0.12.dev2/agentuniverse_product/base/product.py

@@ -0,0 +1,63 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/24 16:29
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: product.py
+from typing import Optional
+
+from agentuniverse.base.component.component_base import ComponentBase
+from agentuniverse.base.component.component_configer_util import ComponentConfigerUtil
+from agentuniverse.base.component.component_enum import ComponentEnum
+from agentuniverse.base.config.application_configer.application_config_manager import ApplicationConfigManager
+from agentuniverse_product.base.product_configer import ProductConfiger
+
+
+class Product(ComponentBase):
+    """The basic class of the product."""
+
+    component_type: ComponentEnum = ComponentEnum.PRODUCT
+    id: Optional[str] = None
+    nickname: Optional[str] = None
+    type: Optional[str] = None
+    avatar: Optional[str] = None
+    _instance: Optional[ComponentBase] = None
+
+    @property
+    def instance(self) -> ComponentBase:
+        return self._instance
+
+    def get_instance_code(self) -> str:
+        """Return the full name of the product."""
+        appname = ApplicationConfigManager().app_configer.base_info_appname
+        return f"{appname}.product.{self.id}"
+
+    def initialize_by_component_configer(self,
+                                         product_configer: ProductConfiger) -> 'Product':
+        """Initialize the Product by the ProductConfiger object.
+
+        Args:
+            product_configer(ProductConfiger): A configer contains product basic info.
+        Returns:
+            Product: A Product instance.
+        """
+        if product_configer.nickname:
+            self.nickname = product_configer.nickname
+        if product_configer.id:
+            self.id = product_configer.id
+        if product_configer.type:
+            self.type = product_configer.type
+        if product_configer.avatar:
+            self.avatar = product_configer.avatar
+        self.init_instance()
+        return self
+
+    def init_instance(self):
+        """Initialize the specific component instance corresponding to the product."""
+        component_manager_clz = ComponentConfigerUtil.get_component_manager_clz_by_type(
+            ComponentEnum.from_value(self.type))
+        self._instance = component_manager_clz().get_instance_obj(self.id)
+        if self._instance is None:
+            raise ValueError(f"The aU instance corresponding to the product id does not exist,"
+                             f" please check the product type and id parameters.")

agentuniverse-0.0.12.dev2/agentuniverse_product/base/product_configer.py

@@ -0,0 +1,91 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/24 16:35
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: product_configer.py
+from typing import Optional
+
+from agentuniverse.base.config.component_configer.component_configer import ComponentConfiger
+from agentuniverse.base.config.configer import Configer
+
+PRODUCT_COMPONENT_TYPE = ['AGENT', 'KNOWLEDGE', 'TOOL', 'PLANNER', 'LLM']
+
+
+class ProductConfiger(ComponentConfiger):
+    """The ProductConfiger class, used to load and manage the product configuration."""
+
+    _ComponentConfiger__metadata_class: Optional[str] = None
+    _ComponentConfiger__metadata_module: Optional[str] = None
+
+    def __init__(self, configer: Optional[Configer] = None):
+        """Initialize the ProductConfiger."""
+        super().__init__(configer)
+        self.__nickname: Optional[str] = None
+        self.__id: Optional[str] = None
+        self.__type: Optional[str] = None
+        self.__avatar: Optional[str] = None
+        self.__set_default_meta_info()
+
+    @property
+    def nickname(self) -> Optional[str]:
+        """Nickname field."""
+        return self.__nickname
+
+    @property
+    def id(self) -> Optional[str]:
+        """ID field."""
+        return self.__id
+
+    @property
+    def type(self) -> Optional[str]:
+        """Type field."""
+        return self.__type
+
+    @property
+    def avatar(self) -> Optional[str]:
+        """Avatar field."""
+        return self.__avatar
+
+    def __set_default_meta_info(self):
+        """Set default instantiated class of product."""
+        if (not hasattr(self, '_ComponentConfiger__metadata_module')
+                or self._ComponentConfiger__metadata_module is None):
+            self._ComponentConfiger__metadata_module = ("agentuniverse_product."
+                                                        "base.product")
+        if (not hasattr(self, '_ComponentConfiger__metadata_class')
+                or self._ComponentConfiger__metadata_class is None):
+            self._ComponentConfiger__metadata_class = 'Product'
+
+    def load(self) -> 'ProductConfiger':
+        """Setting property using own configer member property.
+
+        Returns:
+            ProductConfiger: A ProductConfiger instance.
+        """
+        return self.load_by_configer(self.configer)
+
+    def load_by_configer(self, configer: Configer) -> 'ProductConfiger':
+        """Load the configuration by the Configer object.
+        Args:
+            configer(Configer): the Configer object
+        Returns:
+            ProductConfiger: the ProductConfiger object
+        """
+        super().load_by_configer(configer)
+        self.__set_default_meta_info()
+        try:
+            self.__nickname = configer.value.get('nickname')
+            if self.__nickname is None:
+                raise ValueError(f"Product nickname is required parameter.")
+            self.__id = configer.value.get('id')
+            if self.__id is None:
+                raise ValueError(f"Product id is required parameter.")
+            self.__type = configer.value.get('type')
+            self.__avatar = configer.value.get('avatar')
+            if self.__type not in PRODUCT_COMPONENT_TYPE:
+                raise ValueError(f"Invalid product type: {self.__type}")
+        except Exception as e:
+            raise Exception(f"Failed to parse the Product configuration: {e}")
+        return self

agentuniverse-0.0.12.dev2/agentuniverse_product/base/product_manager.py

@@ -0,0 +1,19 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/24 16:34
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: product_manager.py
+from agentuniverse.base.annotation.singleton import singleton
+from agentuniverse.base.component.component_enum import ComponentEnum
+from agentuniverse.base.component.component_manager_base import ComponentManagerBase
+from agentuniverse_product.base.product import Product
+
+
+@singleton
+class ProductManager(ComponentManagerBase[Product]):
+    """A singleton manager class of the product."""
+
+    def __init__(self):
+        super().__init__(ComponentEnum.PRODUCT)

agentuniverse-0.0.12.dev2/agentuniverse_product/base/util/__init__.py

@@ -0,0 +1,7 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/25 23:33
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: __init__.py.py

agentuniverse-0.0.12.dev2/agentuniverse_product/base/util/yaml_util.py

@@ -0,0 +1,43 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/29 12:14
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: yaml_util.py
+from ruamel.yaml import YAML
+
+
+def update_nested_yaml_value(config_path, updates) -> None:
+    """Update nested values in YAML
+
+    Args:
+        config_path(str): The path to the YAML file.
+
+        updates(dict): A dictionary of key-value pairs to update in the YAML file.
+        The keys should be dot-separated paths to the target values,
+        and the values should be the new values to set.
+    """
+    yaml = YAML()
+    # read an existing YAML file
+    with open(config_path, 'r', encoding='utf-8') as file:
+        config_data = yaml.load(file)
+
+    # batch update the value of the target key
+    for path, new_value in updates.items():
+        keys = path.split('.')
+        d = config_data
+        try:
+            for key in keys[:-1]:
+                if key not in d:
+                    raise KeyError(f"Key '{key}' not found in the configuration")
+                d = d[key]
+            if keys[-1] in d:
+                # modify the value of the target key
+                d[keys[-1]] = new_value
+        except Exception as e:
+            print(f"Skipping update for '{path}': {e}")
+
+    # write the modified content back to the YAML file
+    with open(config_path, 'w', encoding='utf-8') as file:
+        yaml.dump(config_data, file)

agentuniverse-0.0.12.dev2/agentuniverse_product/dal/__init__.py

@@ -0,0 +1,7 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/25 17:42
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: __init__.py.py

agentuniverse-0.0.12.dev2/agentuniverse_product/dal/message_library.py

@@ -0,0 +1,84 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/25 19:34
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: message_library.py
+import datetime
+
+from sqlalchemy import JSON, Integer, String, DateTime, Column, Index, asc
+from sqlalchemy.orm import declarative_base
+
+from agentuniverse.base.annotation.singleton import singleton
+from agentuniverse.database.sqldb_wrapper_manager import SQLDBWrapperManager
+from agentuniverse_product.dal.model.message_do import MessageDO
+
+MESSAGE_TABLE_NAME = 'message'
+Base = declarative_base()
+
+
+class MessageORM(Base):
+    """Sqlalchemy orm Model for message table."""
+    __tablename__ = MESSAGE_TABLE_NAME
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    session_id = Column(String(50), nullable=False)
+    content = Column(JSON)
+    ext_info = Column(JSON)
+    gmt_created = Column(DateTime, default=datetime.datetime.now)
+    gmt_modified = Column(DateTime, default=datetime.datetime.now,
+                          onupdate=datetime.datetime.now)
+    __table_args__ = (
+        Index('message_ix_session_id', 'session_id'),
+    )
+
+
+@singleton
+class MessageLibrary:
+    @staticmethod
+    def get_db_session():
+        """Get the database session."""
+        system_sqldb_wrapper = SQLDBWrapperManager().get_instance_obj('__system_db__')
+        return system_sqldb_wrapper.get_session()()
+
+    def add_message(self, message_do: MessageDO) -> int:
+        """Add a message to the database."""
+        with self.get_db_session() as db_session:
+            message_orm = MessageORM(**message_do.model_dump())
+            db_session.add(message_orm)
+            db_session.commit()
+            return message_orm.id
+
+    def delete_messages(self, session_id: str):
+        """Delete messages from the database using the provided `session_id`."""
+        with self.get_db_session() as db_session:
+            message_orm_list = db_session.query(MessageORM).filter(
+                MessageORM.session_id == session_id)
+            if message_orm_list:
+                for message_orm in message_orm_list:
+                    db_session.delete(message_orm)
+                db_session.commit()
+
+    def get_messages(self, session_id: str) -> list[MessageDO]:
+        """Get messages from the database using the provided `session_id`."""
+        with self.get_db_session() as db_session:
+            message_orm_list = db_session.query(MessageORM).filter(
+                MessageORM.session_id == session_id).order_by(asc(MessageORM.gmt_modified)).all()
+            res = []
+            if message_orm_list:
+                for message_orm in message_orm_list:
+                    res.append(self.__message_orm_to_do(message_orm))
+            return res
+
+    @staticmethod
+    def __message_orm_to_do(message_orm: MessageORM) -> MessageDO:
+        """Convert a MessageORM object to a MessageDO object."""
+        message_do = MessageDO(
+            session_id="",
+            content='',
+            ext_info=dict(),
+        )
+        for column in message_orm.__table__.columns:
+            setattr(message_do, column.name,
+                    getattr(message_orm, column.name))
+        return message_do

agentuniverse-0.0.12.dev2/agentuniverse_product/dal/model/__init__.py

@@ -0,0 +1,7 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/25 21:47
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: __init__.py.py

agentuniverse-0.0.12.dev2/agentuniverse_product/dal/model/message_do.py

@@ -0,0 +1,22 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/25 21:53
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: message_do.py
+import datetime
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class MessageDO(BaseModel):
+    id: Optional[int] = Field(description="ID", default=None)
+    session_id: str = Field(description="Session id")
+    ext_info: Optional[dict] = Field(description="Message ext info.", default={})
+    content: Optional[str] = Field(description="Message content", default='')
+    gmt_created: Optional[datetime.datetime] = Field(
+        description="Create time", default_factory=datetime.datetime.now)
+    gmt_modified: Optional[datetime.datetime] = Field(
+        description="Modified time", default_factory=datetime.datetime.now)

agentuniverse-0.0.12.dev2/agentuniverse_product/dal/model/session_do.py

@@ -0,0 +1,22 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/7/25 21:53
+# @Author  : wangchongshi
+# @Email   : wangchongshi.wcs@antgroup.com
+# @FileName: session_do.py
+import datetime
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class SessionDO(BaseModel):
+    id: Optional[int] = Field(description="ID", default=None)
+    session_id: str = Field(description="Session id")
+    agent_id: str = Field(description="Agent id.")
+    ext_info: Optional[dict] = Field(description="Session ext info.", default={})
+    gmt_created: Optional[datetime.datetime] = Field(
+        description="Create time", default_factory=datetime.datetime.now)
+    gmt_modified: Optional[datetime.datetime] = Field(
+        description="Modified time", default_factory=datetime.datetime.now)