pro-craft 0.1.25__py3-none-any.whl → 0.1.27__py3-none-any.whl

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,8 @@ class Prompt(PromptBase):
                 f"is_deleted='{self.is_deleted}...'>"
                 )
 
-
 class UseCase(PromptBase):
-    __tablename__ = 'usecase' # 数据库中的表名，你可以改成你希望的名字
+    __tablename__ = 'ai_usecase' # 数据库中的表名，你可以改成你希望的名字
 
     # id (int, primary_key=True, autoincrement=True)
     # 你的属性表中 id 为 int, true (not null), true (primary key), 0 (length), ASC (key order), true (auto increment)

pro_craft/prompt_craft/async_.py

@@ -34,6 +34,8 @@ class IntellectRemoveError(Exception):
     pass
 
 BATCH_SIZE = 100
+MIN_SUCCESS_RATE = 00.0 # 这里定义通过阈值, 高于该比例则通过
+
 
 def slog(s, target: str = "target",logger = None):
     COLOR_GREEN = "\033[92m"
@@ -94,7 +96,7 @@ async def get_last_sync_time(target_session: AsyncSession) -> datetime:
     """从目标数据库获取上次同步时间"""
     # 修正点：使用 select() 和 execute()
     result = await target_session.execute(
-        select(SyncMetadata).filter_by(table_name="sync_metadata")
+        select(SyncMetadata).filter_by(table_name="ai_sync_metadata")
     )
     metadata_entry = result.scalar_one_or_none() # 获取单个对象或 None
 
@@ -110,7 +112,7 @@ async def update_last_sync_time(target_session: AsyncSession, new_sync_time: dat
     """更新目标数据库的上次同步时间"""
     # 修正点：使用 select() 和 execute()
     result = await target_session.execute(
-        select(SyncMetadata).filter_by(table_name="sync_metadata")
+        select(SyncMetadata).filter_by(table_name="ai_sync_metadata")
     )
     metadata_entry = result.scalar_one_or_none()
 
@@ -118,7 +120,7 @@ async def update_last_sync_time(target_session: AsyncSession, new_sync_time: dat
         metadata_entry.last_sync_time = new_sync_time
     else:
         # 如果不存在，则创建
-        new_metadata = SyncMetadata(table_name="sync_metadata", last_sync_time=new_sync_time)
+        new_metadata = SyncMetadata(table_name="ai_sync_metadata", last_sync_time=new_sync_time)
         target_session.add(new_metadata)
     
     # 异步提交事务
@@ -128,6 +130,8 @@ async def update_last_sync_time(target_session: AsyncSession, new_sync_time: dat
 
 
 
+
+
 class IntellectType(Enum):
     train = "train"
     inference = "inference"
@@ -155,7 +159,7 @@ class AsyncIntel():
                                     pool_pre_ping=True,  # 使用前检查连接活性
                                     pool_timeout=30      # 等待连接池中连接的最长时间（秒）
                                            )
-        
+        self.create_specific_database(self.engine,["ai_prompts","ai_usecase"])
 
         if model_name in ["gemini-2.5-flash-preview-05-20-nothinking",]:
             self.llm = BianXieAdapter(model_name = model_name)
@@ -165,8 +169,23 @@ class AsyncIntel():
             print('Use BianXieAdapter')
             self.llm = BianXieAdapter()
             
-    async def create_database(self):
-        async with self.engine.begin() as conn:
+    async def create_specific_database(self,engine, tables_to_create_names: list[str]):
+        async with engine.begin() as conn:
+            # 从 metadata 中获取对应的 Table 对象
+            specific_database_objects = []
+            for table_name in tables_to_create_names:
+                if table_name in PromptBase.metadata.tables:
+                    specific_database_objects.append(PromptBase.metadata.tables[table_name])
+                else:
+                    print(f"Warning: Table '{table_name}' not found in metadata.")
+
+            if specific_database_objects:
+                await conn.run_sync(PromptBase.metadata.create_all, tables=specific_database_objects)
+            else:
+                print("No specific tables to create.")
+
+    async def create_database(self,engine):
+        async with engine.begin() as conn:
             await conn.run_sync(PromptBase.metadata.create_all)
             
     async def _get_latest_prompt_version(self,target_prompt_id,session):
@@ -214,9 +233,7 @@ class AsyncIntel():
 
     async def sync_prompt_data_to_database(self,database_url:str):
         target_engine = create_async_engine(database_url, echo=False)
-        async with target_engine.begin() as conn:
-            await conn.run_sync(PromptBase.metadata.create_all)
-            
+        await self.create_database(target_engine) 
         async with create_async_session(self.engine) as source_session:
             async with create_async_session(target_engine) as target_session:
             
@@ -364,6 +381,21 @@ class AsyncIntel():
         session.add(prompt1)
         await session.commit() # 提交事务，将数据写入数据库
 
+    async def get_use_case_by_sql(self,
+                             target_prompt_id: str,
+                             session = None
+                            ):
+        """
+        从sql保存提示词
+        """
+        stmt = select(UseCase).filter(UseCase.is_deleted == 0,
+                                      UseCase.prompt_id == target_prompt_id)
+        
+        result = await session.execute(stmt)
+        # use_case = result.scalars().one_or_none()
+        use_case = result.scalars().all()
+        return use_case
+
     async def save_use_case_by_sql(self,
                              prompt_id: str,
                              use_case:str = "",
@@ -374,6 +406,8 @@ class AsyncIntel():
         """
         从sql保存提示词
         """
+        #TODO 存之前保证数据库中相同的prompt_id中没有重复的use_case
+
         use_case = UseCase(prompt_id=prompt_id, 
                         use_case = use_case,
                         output = output,
@@ -789,7 +823,6 @@ class AsyncIntel():
                 )
 
         try:
-
             json_str = extract_(ai_result,r'json')
             # json_str = fix_broken_json_string(json_str)
             ai_result = json.loads(json_str)
@@ -804,7 +837,7 @@ class AsyncIntel():
                 OutputFormat(**ai_result)
 
             except JSONDecodeError as e:
-                raise IntellectRemoveFormatError(f"prompt_id: {prompt_id} 在生成后做json解析时报错") from e
+                raise IntellectRemoveFormatError(f"prompt_id: {prompt_id} 生成的内容为无法被Json解析 {e}") from e
         
         except ValidationError as e:
             err_info = e.errors()[0]
@@ -845,7 +878,6 @@ class AsyncIntel():
             )
         results = await asyncio.gather(*tasks, return_exceptions=False)
         return results
-
     
     def intellect_remove_warp(self,prompt_id: str):
         def outer_packing(func):
@@ -874,3 +906,90 @@ class AsyncIntel():
                 return result
             return wrapper
         return outer_packing
+
+    async def intellect_remove_format_eval(self,
+                    OutputFormat: object,
+                    prompt_id: str,
+                    ExtraFormats: list[object] = [],
+                    version: str = None,
+                    ):
+        
+        async with create_async_session(self.engine) as session:
+            use_cases = await self.get_use_case_by_sql(target_prompt_id=prompt_id,session=session)
+            prompt_result = await self.get_prompts_from_sql(prompt_id=prompt_id,
+                                                                   session=session)
+            if prompt_result is None:
+                raise IntellectRemoveError("不存在的prompt_id")
+            if prompt_result.action_type != "inference":
+                raise IntellectRemoveError("请在inference模式下使用次类")
+            
+
+            total_assertions = len(use_cases)
+            result_cases = []
+
+            async def evals_func(use_case,prompt_id,OutputFormat,ExtraFormats,version):
+                try:
+                    # 这里将参数传入
+                    await self.intellect_remove_format(
+                        input_data = use_case.use_case,
+                        prompt_id = prompt_id,
+                        OutputFormat = OutputFormat,
+                        ExtraFormats = ExtraFormats,
+                        version = version,
+                        inference_save_case = False,
+                    )
+                    # TODO base_eval
+                    # TODO 人类评价 eval
+                    # TODO llm 评价 eval
+                    result_cases.append({"type":"Successful","case":use_case.use_case,"reply":f"pass"})
+                    use_case.output = "Successful"
+                except IntellectRemoveFormatError as e:
+                    result_cases.append({"type":"FAILED","case":use_case.use_case,"reply":f"{e}"})
+                    use_case.output = f"{"FAILED"}-{e}"
+                except Exception as e: # 捕获其他可能的错误
+                    result_cases.append({"type":"FAILED","case":use_case.use_case,"reply":f"Exp {e}"})
+                    use_case.output = f"{"FAILED"}-{e}"
+                    await session.commit()
+
+            tasks = []
+            for use_case in use_cases:
+                tasks.append(
+                    evals_func(
+                        use_case = use_case,
+                        prompt_id = prompt_id,
+                        OutputFormat = OutputFormat,
+                        ExtraFormats = ExtraFormats,
+                        version = version
+                    )
+                )
+            await asyncio.gather(*tasks, return_exceptions=False)
+
+
+            successful_assertions = 0
+            bad_case = []
+            for i in result_cases:
+                if i['type'] == "Successful":
+                    successful_assertions += 1
+                else:
+                    bad_case.append(i)
+
+            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 >= MIN_SUCCESS_RATE:
+            #     return "通过", json.dumps(result_cases,ensure_ascii=False)
+            # else:
+            #     return "未通过",json.dumps(result_cases,ensure_ascii=False)
+
+            print(bad_case)
+
+            
+            # return results
+
+
+
+
+# 整体测试d, 测试未通过d, 大模型调整再测试, 依旧不通过, 大模型裂变, 仍不通过, 互换人力

pro_craft/prompt_craft/new.py

@@ -37,19 +37,19 @@ def slog(s, target: str = "target",logger = None):
 
 def get_last_sync_time(target_session) -> datetime:
     """从目标数据库获取上次同步时间"""
-    metadata_entry = target_session.query(SyncMetadata).filter_by(table_name="sync_metadata").first()
+    metadata_entry = target_session.query(SyncMetadata).filter_by(table_name="ai_sync_metadata").first()
     if metadata_entry:
         return metadata_entry.last_sync_time
     return datetime(1970, 1, 1) # 默认一个很早的时间
 
 def update_last_sync_time(target_session, new_sync_time: datetime):
     """更新目标数据库的上次同步时间"""
-    metadata_entry = target_session.query(SyncMetadata).filter_by(table_name="sync_metadata").first()
+    metadata_entry = target_session.query(SyncMetadata).filter_by(table_name="ai_sync_metadata").first()
     if metadata_entry:
         metadata_entry.last_sync_time = new_sync_time
     else:
         # 如果不存在，则创建
-        new_metadata = SyncMetadata(table_name="sync_metadata", last_sync_time=new_sync_time)
+        new_metadata = SyncMetadata(table_name="ai_sync_metadata", last_sync_time=new_sync_time)
         target_session.add(new_metadata)
     target_session.commit()
     print(f"Updated last sync time to: {new_sync_time}")

pro_craft/prompt_craft/sync.py

@@ -39,19 +39,19 @@ def slog(s, target: str = "target",logger = None):
 
 def get_last_sync_time(target_session) -> datetime:
     """从目标数据库获取上次同步时间"""
-    metadata_entry = target_session.query(SyncMetadata).filter_by(table_name="sync_metadata").first()
+    metadata_entry = target_session.query(SyncMetadata).filter_by(table_name="ai_sync_metadata").first()
     if metadata_entry:
         return metadata_entry.last_sync_time
     return datetime(1970, 1, 1) # 默认一个很早的时间
 
 def update_last_sync_time(target_session, new_sync_time: datetime):
     """更新目标数据库的上次同步时间"""
-    metadata_entry = target_session.query(SyncMetadata).filter_by(table_name="sync_metadata").first()
+    metadata_entry = target_session.query(SyncMetadata).filter_by(table_name="ai_sync_metadata").first()
     if metadata_entry:
         metadata_entry.last_sync_time = new_sync_time
     else:
         # 如果不存在，则创建
-        new_metadata = SyncMetadata(table_name="sync_metadata", last_sync_time=new_sync_time)
+        new_metadata = SyncMetadata(table_name="ai_sync_metadata", last_sync_time=new_sync_time)
         target_session.add(new_metadata)
     target_session.commit()
     print(f"Updated last sync time to: {new_sync_time}")

pro_craft/server/router/prompt.py

@@ -2,7 +2,7 @@
 
 from fastapi import APIRouter
 from pro_craft import Intel,AsyncIntel
-from pro_craft.utils import create_session
+from pro_craft.utils import create_async_session
 
 def create_router(database_url: str,
                   slave_database_url: str,
@@ -44,7 +44,7 @@ def create_router(database_url: str,
 
     @router.get("/get_latest_prompt")
     async def get_latest_prompt(prompt_id: str):
-        with create_session(intels.engine) as session:
+        async with create_async_session(intels.engine) as session:
             result = await intels.get_prompts_from_sql(
                 prompt_id=prompt_id,
                 session=session
@@ -59,7 +59,7 @@ def create_router(database_url: str,
 
     @router.get("/roll_back")
     async def roll_back(prompt_id:str,version:str):
-        with create_session(intels.engine) as session:
+        async with create_async_session(intels.engine) as session:
             result = await intels.get_prompts_from_sql(
                 prompt_id=prompt_id,
                 version = version,

{pro_craft-0.1.25.dist-info → pro_craft-0.1.27.dist-info}/METADATA

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

{pro_craft-0.1.25.dist-info → pro_craft-0.1.27.dist-info}/RECORD

@@ -1,20 +1,19 @@
 pro_craft/__init__.py,sha256=KU9yexREoJVO5eyDNnOCzdYC8Bfo6p4Z45KWZTeO63U,698
-pro_craft/database.py,sha256=5dV-h9dVaS6euHLCtf0gYfq2pchl2QFdb2PEM4gTEU4,8740
+pro_craft/database.py,sha256=FJmknolcXpx6sYHuiS1LgXX7FV8drSpoN3RwVvRR2FI,8741
 pro_craft/file_manager.py,sha256=abVAlJ07_egWNuTj4JiP4me8NloQrsXGNd-SP63ab94,3738
 pro_craft/log.py,sha256=x9RS_0LITN2SE8dcVaEcUFdcfr__jCYpFkIkVmqE5f0,3061
 pro_craft/utils.py,sha256=R1DFkS4dsm5dIhg8lLTgBBvItvIYyyojROdh-ykqiYk,5250
 pro_craft/code_helper/coder.py,sha256=L6pRQr0pYRIHrMFZ4-pO_tZf1koxgGgF3L7Vl-GIyjM,24687
 pro_craft/code_helper/designer.py,sha256=3gyCqrjcw61sHzDjUPKhL1LOAE8xWLLbNT8NlK2mFLc,4739
 pro_craft/prompt_craft/__init__.py,sha256=83ruWO1Oci-DWvdVhPqcQrgdZTNfbmK72VQCkWASk7A,80
-pro_craft/prompt_craft/async_.py,sha256=xhC6HJCQa1a2u08GFK582d60CDo8e__6vafNCIg2-4U,38689
-pro_craft/prompt_craft/evals.py,sha256=XzaaQgA-Vgo8MNPYTdN0hyFJNmgTvw2jroWmeA02pBs,1847
-pro_craft/prompt_craft/new.py,sha256=cHugfhLNUMQk_l8JQBg5ZQXPYXqspexnCOgp3YUzoD0,25894
-pro_craft/prompt_craft/sync.py,sha256=X2BOzdqLBVZyMm5noEOaX2X5LYHl3nl2d3qAOwLoJr8,26169
+pro_craft/prompt_craft/async_.py,sha256=j_Jy-TNhfFehMT80kOCLelaICU-Ns3OOrPEn4s18Z8k,43846
+pro_craft/prompt_craft/new.py,sha256=ULjGGl95vmHrOs7XECJGlaqj1NE9BypE5WnFYhGugRY,25903
+pro_craft/prompt_craft/sync.py,sha256=w-zXC9_yVxgvnotimcs2POEXaHWD6ibspdPwWL9zqXk,26178
 pro_craft/server/mcp/__init__.py,sha256=4dbl-lFcm0r2tkOP04OxqiZG2jR-rqF181qi2AfU6UA,123
 pro_craft/server/mcp/prompt.py,sha256=OZrsyUfSQMOY_KX7dWthW209adz5JfELsQ0ODfuQR44,1245
 pro_craft/server/router/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pro_craft/server/router/prompt.py,sha256=qguiwCQ6bSsS-mLi3zkGUHcWDgIVOnIMHwSxOt0-S1o,2877
-pro_craft-0.1.25.dist-info/METADATA,sha256=9MKKYmM2kJsF2wHm8F8a8QLzNU8qWBWNwVt8cV2StM8,1689
-pro_craft-0.1.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pro_craft-0.1.25.dist-info/top_level.txt,sha256=yqYDHArnYMWpeCxkmGRwlL6sJtxiOUnYylLDx9EOgFg,10
-pro_craft-0.1.25.dist-info/RECORD,,
+pro_craft/server/router/prompt.py,sha256=Wa4FfYRL6oeyA3F-79pmPeIH0Vo8wSEv7RH1lP6jXck,2907
+pro_craft-0.1.27.dist-info/METADATA,sha256=maDxzIKWmnOpPvjg24-cqnNSOJrAEalQRybw4QHpF0A,1689
+pro_craft-0.1.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pro_craft-0.1.27.dist-info/top_level.txt,sha256=yqYDHArnYMWpeCxkmGRwlL6sJtxiOUnYylLDx9EOgFg,10
+pro_craft-0.1.27.dist-info/RECORD,,

pro_craft/prompt_craft/evals.py

@@ -1,61 +0,0 @@
-
-import json
-
-
-############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
-
-
-
-
-