pro-craft 0.1.22__py3-none-any.whl → 0.1.36__py3-none-any.whl

pro_craft/__init__.py

@@ -1,23 +1,18 @@
+
+
+
 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_ = Log(console_level = logging.WARNING, # 显示控制台的等级
              log_file_name="app.log")
 logger = Log_.logger
-Log_.set_super_log(logger.critical)
-
+Log_.set_super_log(logger.critical) # 控制superlog 打印的等级 默认是最高级单独存储一个文件
 super_log = Log_.super_log # 调试工具
-
-def slog(s, target: str = "target",logger = logger.info):
-    COLOR_GREEN = "\033[92m"
-    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)
+inference_save_case = False
 
 from .prompt_craft import AsyncIntel, Intel, IntelNew

pro_craft/database.py

@@ -72,7 +72,7 @@ PromptBase = declarative_base()
 
 class SyncMetadata(PromptBase):
     """用于存储同步元数据的表模型"""
-    __tablename__ = "sync_metadata"
+    __tablename__ = "ai_sync_metadata"
     id = Column(Integer, primary_key=True, autoincrement=True)
     last_sync_time = Column(DateTime, default=datetime(1970, 1, 1))
     table_name = Column(String(255), unique=True)
@@ -80,9 +80,8 @@ class SyncMetadata(PromptBase):
     def __repr__(self):
         return f"<SyncMetadata(table_name='{self.table_name}', last_sync_time='{self.last_sync_time}')>"
 
-
 class Prompt(PromptBase):
-    __tablename__ = 'prompts_table' # 数据库中的表名，你可以改成你希望的名字
+    __tablename__ = 'ai_prompts' # 数据库中的表名，你可以改成你希望的名字
     # __tablename__ = 'llm_prompt' # 数据库中的表名，你可以改成你希望的名字
 
     # 定义联合唯一约束
@@ -177,9 +176,13 @@ class Prompt(PromptBase):
                 f"is_deleted='{self.is_deleted}...'>"
                 )
 
-
 class UseCase(PromptBase):
-    __tablename__ = 'usecase' # 数据库中的表名，你可以改成你希望的名字
+    __tablename__ = 'ai_usecase' # 数据库中的表名，你可以改成你希望的名字
+
+    # __table_args__ = (
+    #     UniqueConstraint('prompt_id', 'use_case', name='_prompt_id_version_uc'),
+    #     # 'name' 参数是可选的，用于给数据库中的约束指定一个名称，方便管理和调试
+    # )
 
     # id (int, primary_key=True, autoincrement=True)
     # 你的属性表中 id 为 int, true (not null), true (primary key), 0 (length), ASC (key order), true (auto increment)
@@ -208,6 +211,14 @@ class UseCase(PromptBase):
         comment="用例"
     )
 
+    timestamp = Column(
+        DateTime,
+        nullable=False,      # 不能为空
+        server_default=text('CURRENT_TIMESTAMP'),
+        onupdate=text('CURRENT_TIMESTAMP'),
+        comment="时间戳"
+    )
+
     output = Column(
         Text,               # TEXT 类型，适用于长文本
         nullable=False,     # 不能为空
@@ -222,6 +233,12 @@ class UseCase(PromptBase):
         comment="理想结果"      # 列注释
     )
 
+    faired_time = Column(
+        Integer,
+        nullable=False,     # 不能为空
+        comment="失败次数"
+    )
+
     is_deleted = Column(Boolean, default=False, server_default=text('0')) 
 
     # 定义 __repr__ 方法以便打印对象时有清晰的表示

pro_craft/file_manager.py

@@ -7,7 +7,7 @@ from pro_craft import Intel
 from enum import Enum
 import json
 
-from pro_craft.database import Base, Prompt
+from pro_craft.database import Prompt
 from sqlalchemy import create_engine, Column, Integer, String
 from sqlalchemy.orm import sessionmaker, declarative_base
 from pro_craft.utils import create_session
@@ -30,7 +30,7 @@ class ContentManager():
         self.qdrant = QdrantManager(host = "localhost")
         self.neo = None
 
-    
+
     @intel.intellect_2(IntellectType.inference,
                     prompt_id = "db_help_001",
                     demand="""

pro_craft/log.py

@@ -12,6 +12,7 @@ class Log:
     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
+        os.makedirs("logs", exist_ok=False)
         self.LOG_FILE_PATH = os.path.join("logs", log_file_name)
         self.logger = self.get_logger()
         self.super_log_level = self.logger.critical
@@ -60,23 +61,22 @@ class Log:
             )
             file_handler_debug.setLevel(logging.WARNING)  # 文件中显示所有指定级别的日志
             file_handler_debug.setFormatter(formatter)
-            logger.addHandler(file_handler_debug)
+
+            file_handler_cri = RotatingFileHandler(
+                self.LOG_FILE_PATH.replace('.log','_slog.log'),
+                maxBytes=5 * 1024 * 1024,  # 10 MB
+                backupCount=5,
+                encoding="utf-8",
+            )
+            file_handler_cri.setLevel(logging.CRITICAL)  # 文件中显示所有指定级别的日志
+            file_handler_cri.setFormatter(formatter)
+            logger.addHandler(file_handler_cri)
         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)
+    def super_log(self,s, target: str = "target",logger = None):
+        logger = logger or self.super_log_level
+        logger("\n" + "=="*25 + target +"=="*25 +   f"\n   type:  {str(type(s))}" + f"\ncontent:  {s}")