pro-craft 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

pro_craft/__init__.py

@@ -0,0 +1,25 @@
+from dotenv import load_dotenv, find_dotenv
+dotenv_path = find_dotenv()
+load_dotenv(".env", override=True)
+
+from .log import Log
+import logging
+Log_ = Log(console_level = logging.WARNING, 
+             log_file_name="app.log")
+logger = Log_.logger
+Log_.set_super_log(logger.critical)
+
+super_log = Log_.super_log # 调试工具
+
+def slog(s, target: str = "target",logger = logger.info):
+    COLOR_RED = "\033[91m"
+    COLOR_GREEN = "\033[92m"
+    COLOR_YELLOW = "\033[93m"
+    COLOR_BLUE = "\033[94m"
+    COLOR_RESET = "\033[0m" # 重置颜色
+
+    logger("\n"+f"{COLOR_GREEN}=={COLOR_RESET}" * 50)
+    logger(target + "\n       "+"--" * 40)
+    logger(type(s))
+    logger(s)
+    logger("\n"+f"{COLOR_GREEN}=={COLOR_RESET}" * 50)

pro_craft/log.py

@@ -2,117 +2,81 @@ import logging
 import os
 from logging.handlers import RotatingFileHandler, TimedRotatingFileHandler
 
-class Logger:
-    """log
-
-    Returns:
-        _type_: single
-    """
+class Log:
     _instance = None
-
     def __new__(cls, *args, **kwargs):
         if cls._instance is None:
             cls._instance = super().__new__(cls)
         return cls._instance
 
-    def __init__(self,level = 'debug',log_file_name = "app.log"):
-        """
-        Level (级别): 定义日志的严重程度。从低到高依次是：
-        DEBUG: 详细的调试信息，通常只在开发阶段使用。
-        INFO: 确认程序按预期运行。
-        WARNING: 发生了意外，或将来可能出现问题，但程序仍在正常运行。
-        ERROR: 发生了严重错误，程序可能无法执行某些功能。
-        CRITICAL: 发生了严重错误，程序可能无法继续运行。
-                # --- 1. 定义日志级别 ---
-        # 建议在开发环境使用 DEBUG，生产环境使用 INFO 或 WARNING
-
-        """
-        if not hasattr(self, 'initialized'):
-            self.initialized = True
-
-            if level == 'debug':
-                self.LOG_LEVEL = logging.DEBUG  # 开发阶段
-            elif level == 'info':
-                self.LOG_LEVEL = logging.INFO  # 生产阶段
-            elif level == 'warning':
-                self.LOG_LEVEL = logging.WARNING
-            elif level == 'error':
-                self.LOG_LEVEL = logging.ERROR
-            elif level == 'critical':
-                self.LOG_LEVEL = logging.CRITICAL
-            else:
-                self.LOG_LEVEL = logging.INFO  # 默认级别
-
-            # --- 2. 定义日志文件路径和名称 ---
-            self.LOG_DIR = "logs"
-            self.LOG_FILE_NAME = log_file_name
-            self.LOG_FILE_PATH = os.path.join(self.LOG_DIR, self.LOG_FILE_NAME)
-
-            # 确保日志目录存在
-            os.makedirs(self.LOG_DIR, exist_ok=True)
-            self.logger = None
-            self.setup_logging()
-            self.env = 'dev'
-
-    def reset_level(self,level = 'debug',env = 'dev'):
-        if level == 'debug':
-            self.LOG_LEVEL = logging.DEBUG  # 开发阶段
-        elif level == 'info':
-            self.LOG_LEVEL = logging.INFO  # 生产阶段
-        elif level == 'warning':
-            self.LOG_LEVEL = logging.WARNING
-        elif level == 'error':
-            self.LOG_LEVEL = logging.ERROR
-        elif level == 'critical':
-            self.LOG_LEVEL = logging.CRITICAL
-        else:
-            self.LOG_LEVEL = logging.INFO  # 默认级别
-
-        self.setup_logging()
-        self.env = env
+    def __init__(self, console_level = logging.INFO, log_file_name="app.log"):
+        self.Console_LOG_LEVEL = console_level
+        self.log_file_name = log_file_name
+        self.LOG_FILE_PATH = os.path.join("logs", log_file_name)
+        self.logger = self.get_logger()
+        self.super_log_level = self.logger.critical
 
-    def setup_logging(self):
-        """# --- 3. 配置 Logger ---
-        """
-        # 获取根 Logger (也可以创建自定义的 Logger: logging.getLogger('my_app'))
+    def get_logger(self):
         logger = logging.getLogger()
-        logger.setLevel(self.LOG_LEVEL)
-
-        # 避免重复添加 Handler (如果多次调用 setup_logging)
+        # logger.setLevel(self.LOG_LEVEL)
         if not logger.handlers:
             # --- 4. 配置 Formatter (格式化器) ---
-            # 常见格式：时间 - 日志级别 - Logger名称 - 模块名 - 行号 - 消息
             formatter = logging.Formatter(
-                '%(asctime)s - %(levelname)s - %(name)s - %(module)s:%(lineno)d - %(message)s'
+                "%(asctime)s - %(levelname)s - %(name)s - %(funcName)s - %(module)s:%(lineno)d - %(message)s"
             )
-
             # --- 5. 配置 Handler (处理器) ---
 
             # 5.1 控制台处理器 (StreamHandler)
             console_handler = logging.StreamHandler()
-            console_handler.setLevel(logging.INFO) # 控制台只显示 INFO 及以上级别的日志
+            console_handler.setLevel(self.Console_LOG_LEVEL)  # 控制台只显示 INFO 及以上级别的日志
             console_handler.setFormatter(formatter)
             logger.addHandler(console_handler)
 
-            # 5.2 文件处理器 (RotatingFileHandler 或 TimedRotatingFileHandler)
 
+            # 文件
+
+            # 5.2 信息数据
+
+            # RotatingFileHandler: 按文件大小轮转
+            # maxBytes: 单个日志文件的最大字节数 (例如 10MB)
+            # backupCount: 保留的旧日志文件数量
             file_handler = RotatingFileHandler(
                 self.LOG_FILE_PATH,
-                maxBytes=10 * 1024 * 1024, # 10 MB
+                maxBytes=10 * 1024 * 1024,  # 10 MB
                 backupCount=5,
-                encoding='utf-8'
+                encoding="utf-8",
             )
-            file_handler.setLevel(self.LOG_LEVEL) # 文件中显示所有指定级别的日志
+            file_handler.setLevel(logging.INFO)  # 文件中显示所有指定级别的日志
             file_handler.setFormatter(formatter)
             logger.addHandler(file_handler)
 
-        self.logger = logger
+            # 5.3 错误警告日志
 
+            file_handler_debug = RotatingFileHandler(
+                self.LOG_FILE_PATH.replace('.log','_err.log'),
+                maxBytes=10 * 1024 * 1024,  # 10 MB
+                backupCount=5,
+                encoding="utf-8",
+            )
+            file_handler_debug.setLevel(logging.WARNING)  # 文件中显示所有指定级别的日志
+            file_handler_debug.setFormatter(formatter)
+            logger.addHandler(file_handler_debug)
+        return logger
+    
+    def set_super_log(self,logger_info):
+        self.super_log_level = logger_info
+
+    def super_log(self,s, target: str = "target"):
+        COLOR_RED = "\033[91m"
+        COLOR_GREEN = "\033[92m"
+        COLOR_YELLOW = "\033[93m"
+        COLOR_BLUE = "\033[94m"
+        COLOR_RESET = "\033[0m" # 重置颜色
+        log_ = self.super_log_level
+
+        log_("\n"+f"{COLOR_GREEN}=={COLOR_RESET}" * 50)
+        log_(target + "\n       "+"--" * 40)
+        log_(type(s))
+        log_(s)
+        log_("\n"+f"{COLOR_GREEN}=={COLOR_RESET}" * 50)
 
-Log = Logger(log_file_name = "app.log")
-del Logger
-
-"""
-from .log import Log
-logger = Log.logger
-"""

pro_craft/server/mcp/weather.py

@@ -1,22 +1,46 @@
 from datetime import datetime
 from mcp.server.fastmcp import FastMCP
 
-# region MCP Weather
-mcp = FastMCP("Weather")
+from pro_craft.prompt_helper import IntellectType, Intel
 
-@mcp.tool()
-def get_weather(location: str) -> str:
-    return "Cloudy"
+def create_mcp(database_url: str,
+                  slave_database_url: str,
+                  model_name: str):
+    # region MCP Weather
+    mcp = FastMCP("PromptManager")
 
-@mcp.tool()
-def get_time() -> str:
-    return datetime.now().strftime('%Y-%m-%d %H:%M:%S')
-# endregion
+    intels = Intel(
+        database_url=database_url,
+        model_name=model_name
+        )
 
+    @mcp.tool()
+    def push_order(demand: str, prompt_id: str, action_type: str = "train") -> str:
+        result = intels.push_action_order(
+            demand=demand,
+            prompt_id=prompt_id,
+            action_type=action_type
+        )
+        return {"message": "success", "result": result}
+
+    @mcp.tool()
+    def get_latest_prompt(prompt_id: str) -> str:
+        with create_session(intels.engine) as session:
+            result = intels.get_prompts_from_sql(
+                prompt_id=prompt_id,
+                session=session
+            )
+        return {"message": "success", "result": result}
 
-if __name__ == "__main__":
-    mcp.run(transport="streamable-http")
-    # mcp.run(transport="sse")
-    # search_mcp.run(transport="sse", mount_path="/search")
 
+    @mcp.tool()
+    def sync_database() -> str:
+        result = intels.sync_prompt_data_to_database(slave_database_url)
+        return {"message": "success","result":result}
+    
+    return mcp
 
+
+if __name__ == "__main__":
+    mcp = create_mcp()
+    mcp.run(transport="streamable-http")

pro_craft/server/router/prompt.py

@@ -0,0 +1,56 @@
+
+
+from fastapi import APIRouter
+from pro_craft.prompt_helper import IntellectType,Intel
+from pro_craft.utils import create_session
+import os
+
+def create_router(database_url: str,
+                  slave_database_url: str,
+                  model_name: str):
+    """
+    创建一个包含 ProCraft 路由的 FastAPI APIRouter 实例。
+
+    Args:
+        database_url (str): 数据库连接字符串。
+        model_name (str): 用于 Intel 实例的模型名称。
+        api_key_secret (str, optional): 用于验证 API Key 的秘密字符串。
+                                        如果提供，它将覆盖环境变量 PRO_CRAFT_API_KEY。
+                                        如果都不提供，会使用硬编码的 'your_default_secret_key'。
+    Returns:
+        APIRouter: 配置好的 FastAPI APIRouter 实例。
+    """
+
+    intels = Intel(
+        database_url=database_url,
+        model_name=model_name
+        )
+
+    router = APIRouter(
+        tags=["prompt"] # 这里使用 Depends 确保每次请求都验证
+    )
+
+    @router.get("/push_order")
+    async def push_order(demand: str, prompt_id: str, action_type: str = "train"):
+        result = intels.push_action_order(
+            demand=demand,
+            prompt_id=prompt_id,
+            action_type=action_type
+        )
+        return {"message": "success", "result": result}
+
+    @router.get("/get_latest_prompt")
+    async def get_latest_prompt(prompt_id: str):
+        with create_session(intels.engine) as session:
+            result = intels.get_prompts_from_sql(
+                prompt_id=prompt_id,
+                session=session
+            )
+        return {"message": "success", "result": result}
+
+    @router.get("/sync_database")
+    async def sync_database():
+        result = intels.sync_prompt_data_to_database(slave_database_url)
+        return {"message": "success","result":result}
+    
+    return router

pro_craft/utils.py

@@ -14,6 +14,11 @@ import zlib
 from volcenginesdkarkruntime import Ark
 import os
 
+from contextlib import contextmanager
+from sqlalchemy.orm import declarative_base, sessionmaker
+from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine # 异步核心
+from contextlib import asynccontextmanager # 注意这里是 asynccontextmanager
+
 
 def extract_(text: str, pattern_key = r"json",multi = False):
     pattern = r"```"+ pattern_key + r"([\s\S]*?)```"
@@ -96,22 +101,6 @@ def load_inpackage_file(package_name:str, file_name:str,file_type = 'yaml'):
             return f.read()
 
 
-def super_print(s,target:str):
-    print()
-    print()
-    print("=="*21 + target + "=="*21)
-    print()
-    print("=="*50)
-    print(type(s))
-    print("=="*50)
-    print(s)
-    print("=="*50)
-    print()
-
-
-from sqlalchemy.orm import sessionmaker
-
-from contextlib import contextmanager
 @contextmanager
 def create_session(engine):
     # 5. 创建会话 (Session)
@@ -127,17 +116,6 @@ def create_session(engine):
     finally:
         session.close() # 关闭会话，释放资源
 
-from contextlib import contextmanager
-from sqlalchemy import create_engine, Column, Integer, String, UniqueConstraint
-from sqlalchemy.orm import declarative_base, sessionmaker
-from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine # 异步核心
-import asyncio
-
-from contextlib import contextmanager
-
-from contextlib import asynccontextmanager # 注意这里是 asynccontextmanager
-import asyncio
-
 
 @asynccontextmanager
 async def create_async_session(async_engine):

{pro_craft-0.1.12.dist-info → pro_craft-0.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pro-craft
-Version: 0.1.12
+Version: 0.1.14
 Summary: Add your description here
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

pro_craft-0.1.14.dist-info/RECORD

@@ -0,0 +1,15 @@
+pro_craft/__init__.py,sha256=_jw5VKpBAJkqaW9qvjKUGgV7YVTY3HxgZj6R2D1garM,729
+pro_craft/database.py,sha256=5dV-h9dVaS6euHLCtf0gYfq2pchl2QFdb2PEM4gTEU4,8740
+pro_craft/file_manager.py,sha256=2j7lCt9L4mtvAy8_76ibTthXLwKKmVatWIB3DSvQM7U,3805
+pro_craft/log.py,sha256=x9RS_0LITN2SE8dcVaEcUFdcfr__jCYpFkIkVmqE5f0,3061
+pro_craft/utils.py,sha256=R1DFkS4dsm5dIhg8lLTgBBvItvIYyyojROdh-ykqiYk,5250
+pro_craft/code_helper/coder.py,sha256=NXglF1KiPtGe4HZN0MZvFJ8p9Iyd5kzIt72DQGgRwXA,24715
+pro_craft/code_helper/designer.py,sha256=3gyCqrjcw61sHzDjUPKhL1LOAE8xWLLbNT8NlK2mFLc,4739
+pro_craft/server/mcp/__init__.py,sha256=4dbl-lFcm0r2tkOP04OxqiZG2jR-rqF181qi2AfU6UA,123
+pro_craft/server/mcp/weather.py,sha256=46mwcRS-6DFHyOSirgphkpb0Bmtt8Dhx-xHjitQ0FWo,1304
+pro_craft/server/router/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pro_craft/server/router/prompt.py,sha256=VbxuvXvxJ-wLEmMnk7ec3bsIF-WB_JGh7GIYxZkXyWc,1914
+pro_craft-0.1.14.dist-info/METADATA,sha256=L9SmOtG2pHoqsWo150xmnLE_2ThtdKtcFicPFcyoV1s,1800
+pro_craft-0.1.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pro_craft-0.1.14.dist-info/top_level.txt,sha256=yqYDHArnYMWpeCxkmGRwlL6sJtxiOUnYylLDx9EOgFg,10
+pro_craft-0.1.14.dist-info/RECORD,,

pro_craft/evals.py

@@ -1,68 +0,0 @@
-
-import json
-
-from pro_craft.log import Log
-logger = Log.logger
-editing_log = logger.debug
-
-
-
-
-
-############evals##############
-
- # 异步
-class Base_Evals():
-    def __init__(self):
-        """
-        # TODO 2 自动优化prompt 并提升稳定性, 并测试
-        通过重写继承来使用它
-        """
-        self.MIN_SUCCESS_RATE = 00.0 # 这里定义通过阈值, 高于该比例则通过
-
-
-    async def _assert_eval_function(self,params):
-        #这里定义函数的评价体系
-        print(params,'params')
-
-    async def get_success_rate(self,test_cases:list[tuple]):
-        """
-                # 这里定义数据
-
-        """
-
-        successful_assertions = 0
-        total_assertions = len(test_cases)
-        result_cases = []
-
-        for i, params in enumerate(test_cases):
-            try:
-                # 这里将参数传入
-                await self._assert_eval_function(params)
-                successful_assertions += 1
-                result_cases.append({"type":"Successful","--input--":params,"evaluate_info":f"满足要求"})
-            except AssertionError as e:
-                result_cases.append({"type":"FAILED","--input--":params,"evaluate_info":f"ERROR {e}"})
-            except Exception as e: # 捕获其他可能的错误
-                result_cases.append({"type":"FAILED","--input--":params,"evaluate_info":f"ERROR {e}"})
-
-
-        success_rate = (successful_assertions / total_assertions) * 100
-        print(f"\n--- Aggregated Results ---")
-        print(f"Total test cases: {total_assertions}")
-        print(f"Successful cases: {successful_assertions}")
-        print(f"Success Rate: {success_rate:.2f}%")
-
-        if success_rate >= self.MIN_SUCCESS_RATE:
-            return "通过", json.dumps(result_cases,ensure_ascii=False)
-        else:
-            return "未通过",json.dumps(result_cases,ensure_ascii=False)
-
-
-    def global_evals():
-        pass
-
-
-
-
-