beswarm 0.1.12__tar.gz → 0.1.13__tar.gz

{beswarm-0.1.12 → beswarm-0.1.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.12
+Version: 0.1.13
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

beswarm-0.1.13/beswarm/aient/main.py

@@ -0,0 +1,50 @@
+import os
+from datetime import datetime
+
+from src.aient.utils import prompt
+from src.aient.models import chatgpt, claude3, gemini, groq
+LANGUAGE = os.environ.get('LANGUAGE', 'Simplified Chinese')
+GPT_ENGINE = os.environ.get('GPT_ENGINE', 'gpt-4-turbo-2024-04-09')
+
+API = os.environ.get('API', None)
+API_URL = os.environ.get('API_URL', None)
+
+CLAUDE_API = os.environ.get('CLAUDE_API', None)
+GOOGLE_AI_API_KEY = os.environ.get('GOOGLE_AI_API_KEY', None)
+GROQ_API_KEY = os.environ.get('GROQ_API_KEY', None)
+
+current_date = datetime.now()
+Current_Date = current_date.strftime("%Y-%m-%d")
+
+message = "https://arxiv.org/abs/2404.02041 这篇论文讲了啥？"
+systemprompt = os.environ.get('SYSTEMPROMPT', prompt.chatgpt_system_prompt)
+# systemprompt = os.environ.get('SYSTEMPROMPT', prompt.system_prompt.format(LANGUAGE, Current_Date))
+# systemprompt = (
+#     "你是一位旅行专家。你可以规划旅行行程，如果用户有预算限制，还需要查询机票价格。结合用户的出行时间，给出合理的行程安排。"
+#     "在规划行程之前，必须先查找旅行攻略搜索景点信息，即使用 get_city_tarvel_info 查询景点信息。查询攻略后，你需要分析用户个性化需求给出合理的行程安排。充分考虑用户的年龄，情侣，家庭，朋友，儿童，独自旅行等情况。"
+#     "你需要根据用户给出的地点和预算，给出真实准确的行程，包括游玩时长、景点之间的交通方式和移动距离，每天都要给出总的游玩时间。"
+#     "给用户介绍景点的时候，根据查到的景点介绍结合你自己的知识，景点介绍尽量丰富精彩，吸引用户眼球，不要直接复述查到的景点介绍。"
+#     "尽量排满用户的行程，不要有太多空闲时间。"
+#     "你还可以根据用户的需求，给出一些旅行建议。"
+# )
+bot = chatgpt(api_key=API, api_url=API_URL , engine=GPT_ENGINE, system_prompt=systemprompt)
+# bot = claude3(api_key=CLAUDE_API, engine=GPT_ENGINE, system_prompt=systemprompt)
+# bot = gemini(api_key=GOOGLE_AI_API_KEY, engine=GPT_ENGINE, system_prompt=systemprompt)
+# bot = groq(api_key=GROQ_API_KEY, engine=GPT_ENGINE, system_prompt=systemprompt)
+for text in bot.ask_stream(message):
+# for text in bot.ask_stream("今天的微博热搜有哪些？"):
+# for text in bot.ask_stream("250m usd = cny"):
+# for text in bot.ask_stream("我在广州市，想周一去香港，周四早上回来，是去游玩，请你帮我规划整个行程。包括细节，如交通，住宿，餐饮，价格，等等，最好细节到每天各个部分的时间，花费，等等，尽量具体，用户一看就能直接执行的那种"):
+# for text in bot.ask_stream("英伟达最早支持杜比视界的显卡是哪一代"):
+# for text in bot.ask_stream("100个斐波纳切数列的和是多少"):
+# for text in bot.ask_stream("上海有哪些好玩的地方？"):
+# for text in bot.ask_stream("https://arxiv.org/abs/2404.02041 这篇论文讲了啥？"):
+# for text in bot.ask_stream("今天伊朗总统目前的情况怎么样？"):
+# for text in bot.ask_stream("我不是很懂y[..., 2]，y[..., 2] - y[:, 0:1, 0:1, 2]，y[:, 0:1, 0:1, 2]这些对张量的slice操作，给我一些练习demo代码，专门给我巩固这些张量复杂操作。让我从易到难理解透彻所有这样类型的张量操作。"):
+# for text in bot.ask_stream("just say test"):
+# for text in bot.ask_stream("画一只猫猫"):
+# for text in bot.ask_stream("我在上海想去重庆旅游，我只有2000元预算，我想在重庆玩一周，你能帮我规划一下吗？"):
+# for text in bot.ask_stream("我在上海想去重庆旅游，我有一天的时间。你能帮我规划一下吗？"):
+    print(text, end="")
+
+# print("\n bot tokens usage", bot.tokens_usage)

beswarm-0.1.13/beswarm/aient/setup.py

@@ -0,0 +1,15 @@
+from pathlib import Path
+
+from setuptools import setup, find_packages
+
+setup(
+    name="aient",
+    version="1.0.73",
+    description="Aient: The Awakening of Agent.",
+    long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
+    long_description_content_type="text/markdown",
+    packages=find_packages("src"),
+    package_dir={"": "src"},
+    install_requires=Path.open(Path("requirements.txt"), encoding="utf-8").read().splitlines(),
+    include_package_data=True,
+)

beswarm-0.1.13/beswarm/aient/src/aient/__init__.py

@@ -0,0 +1 @@
+from .models import *

beswarm-0.1.13/beswarm/aient/src/aient/core/__init__.py

@@ -0,0 +1 @@
+# core package initialization file

beswarm-0.1.13/beswarm/aient/src/aient/core/log_config.py

@@ -0,0 +1,6 @@
+import logging
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+logger = logging.getLogger("uni-api")
+
+logging.getLogger("httpx").setLevel(logging.CRITICAL)
+logging.getLogger("watchfiles.main").setLevel(logging.CRITICAL)

beswarm-0.1.13/beswarm/aient/src/aient/core/models.py

@@ -0,0 +1,232 @@
+from io import IOBase
+from pydantic import BaseModel, Field, model_validator, ConfigDict
+from typing import List, Dict, Optional, Union, Tuple, Literal, Any
+
+class FunctionParameter(BaseModel):
+    type: str
+    properties: Dict[str, Dict[str, Any]]
+    required: List[str] = None
+    defs: Dict[str, Any] = Field(default=None, alias="$defs")
+
+    class Config:
+        population_by_name = True
+
+class Function(BaseModel):
+    name: str
+    description: str = Field(default=None)
+    parameters: Optional[FunctionParameter] = Field(default=None, exclude=None)
+
+class Tool(BaseModel):
+    type: str
+    function: Function
+
+    @classmethod
+    def parse_raw(cls, json_str: str) -> 'Tool':
+        """从JSON字符串解析Tool对象"""
+        return cls.model_validate_json(json_str)
+
+class FunctionCall(BaseModel):
+    name: str
+    arguments: str
+
+class ToolCall(BaseModel):
+    id: str
+    type: str
+    function: FunctionCall
+
+class ImageUrl(BaseModel):
+    url: str
+
+class ContentItem(BaseModel):
+    type: str
+    text: Optional[str] = None
+    image_url: Optional[ImageUrl] = None
+
+class Message(BaseModel):
+    role: str
+    name: Optional[str] = None
+    arguments: Optional[str] = None
+    content: Optional[Union[str, List[ContentItem]]] = None
+    tool_calls: Optional[List[ToolCall]] = None
+
+class Message(BaseModel):
+    role: str
+    name: Optional[str] = None
+    content: Optional[Union[str, List[ContentItem]]] = None
+    tool_calls: Optional[List[ToolCall]] = None
+    tool_call_id: Optional[str] = None
+
+    class Config:
+        extra = "allow"  # 允许额外的字段
+
+class FunctionChoice(BaseModel):
+    name: str
+
+class ToolChoice(BaseModel):
+    type: str
+    function: Optional[FunctionChoice] = None
+
+class BaseRequest(BaseModel):
+    request_type: Optional[Literal["chat", "image", "audio", "moderation"]] = Field(default=None, exclude=True)
+
+import warnings
+warnings.filterwarnings("ignore", category=UserWarning, message=".*shadows an attribute.*")
+
+class JsonSchema(BaseModel):
+    name: str
+    schema: Dict[str, Any] = Field(validation_alias='schema')
+
+    model_config = ConfigDict(protected_namespaces=())
+
+class ResponseFormat(BaseModel):
+    type: Literal["text", "json_object", "json_schema"]
+    json_schema: Optional[JsonSchema] = None
+
+class Thinking(BaseModel):
+    budget_tokens: Optional[int] = None
+    type: Optional[Literal["enabled", "disabled"]] = None
+
+class StreamOptions(BaseModel):
+    include_usage: Optional[bool] = None
+
+class RequestModel(BaseRequest):
+    model: str
+    messages: List[Message]
+    logprobs: Optional[bool] = None
+    top_logprobs: Optional[int] = None
+    stream: Optional[bool] = None
+    include_usage: Optional[bool] = None
+    temperature: Optional[float] = 0.5
+    top_p: Optional[float] = 1.0
+    max_tokens: Optional[int] = None
+    max_completion_tokens: Optional[int] = None
+    presence_penalty: Optional[float] = 0.0
+    frequency_penalty: Optional[float] = 0.0
+    n: Optional[int] = 1
+    user: Optional[str] = None
+    tool_choice: Optional[Union[str, ToolChoice]] = None
+    tools: Optional[List[Tool]] = None
+    response_format: Optional[ResponseFormat] = None
+    thinking: Optional[Thinking] = None
+    stream_options: Optional[StreamOptions] = None
+
+    def get_last_text_message(self) -> Optional[str]:
+        for message in reversed(self.messages):
+            if message.content:
+                if isinstance(message.content, str):
+                    return message.content
+                elif isinstance(message.content, list):
+                    for item in reversed(message.content):
+                        if item.type == "text" and item.text:
+                            return item.text
+        return ""
+
+    def model_dump(self, **kwargs):
+        data = super().model_dump(**kwargs)
+
+        # 检查并处理 tools 字段
+        if 'tools' in data and data['tools']:
+            for tool in data['tools']:
+                if 'function' in tool:
+                    function_data = tool['function']
+                    # 如果 parameters 为空或没有 properties，则移除
+                    if 'parameters' in function_data and (
+                        function_data['parameters'] is None or
+                        not function_data['parameters'].get('properties')
+                    ):
+                        function_data.pop('parameters', None)
+
+        return data
+
+class ImageGenerationRequest(BaseRequest):
+    prompt: str
+    model: Optional[str] = "dall-e-3"
+    n:  Optional[int] = 1
+    response_format: Optional[str] = "url"
+    size: Optional[str] = "1024x1024"
+    stream: bool = False
+
+class EmbeddingRequest(BaseRequest):
+    input: Union[str, List[Union[str, int, List[int]]]]  # 支持字符串或数组
+    model: str
+    encoding_format: Optional[str] = "float"
+    dimensions: Optional[int] = None
+    user: Optional[str] = None
+    stream: bool = False
+
+class AudioTranscriptionRequest(BaseRequest):
+    file: Tuple[str, IOBase, str]
+    model: str
+    language: Optional[str] = None
+    prompt: Optional[str] = None
+    response_format: Optional[str] = None
+    temperature: Optional[float] = None
+    stream: bool = False
+    timestamp_granularities: Optional[List[str]] = Field(default=["segment"])
+
+    class Config:
+        arbitrary_types_allowed = True
+
+class ModerationRequest(BaseRequest):
+    input: Union[str, List[str]]
+    model: Optional[str] = "text-moderation-latest"
+    stream: bool = False
+
+class TextToSpeechRequest(BaseRequest):
+    model: str
+    input: str
+    voice: str
+    response_format: Optional[str] = "mp3"
+    speed: Optional[float] = 1.0
+    stream: Optional[bool] = False  # Add this line
+
+class UnifiedRequest(BaseModel):
+    data: Union[RequestModel, ImageGenerationRequest, AudioTranscriptionRequest, ModerationRequest, EmbeddingRequest, TextToSpeechRequest]
+
+    @model_validator(mode='before')
+    @classmethod
+    def set_request_type(cls, values):
+        if isinstance(values, dict):
+            if "messages" in values:
+                values["data"] = RequestModel(**values)
+                values["data"].request_type = "chat"
+            elif "prompt" in values:
+                values["data"] = ImageGenerationRequest(**values)
+                values["data"].request_type = "image"
+            elif "file" in values:
+                values["data"] = AudioTranscriptionRequest(**values)
+                values["data"].request_type = "audio"
+            elif "tts" in values.get("model", ""):
+                values["data"] = TextToSpeechRequest(**values)
+                values["data"].request_type = "tts"
+            elif "text-embedding" in values.get("model", ""):
+                values["data"] = EmbeddingRequest(**values)
+                values["data"].request_type = "embedding"
+            elif "input" in values:
+                values["data"] = ModerationRequest(**values)
+                values["data"].request_type = "moderation"
+            else:
+                raise ValueError("无法确定请求类型")
+        return values
+
+if __name__ == "__main__":
+    # 示例JSON字符串
+    json_str = '''
+    {
+        "type": "function",
+        "function": {
+            "name": "clock-time____getCurrentTime____standalone",
+            "description": "获取当前时间",
+            "parameters": {
+                "type": "object",
+                "properties": {}
+            }
+        }
+    }
+    '''
+
+    # 解析JSON字符串为Tool对象
+    tool = Tool.parse_raw(json_str)
+
+    # parameters 字段将被自动排除
+    print(tool.model_dump(exclude_unset=True))