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

pro_craft/database.py

@@ -179,6 +179,11 @@ class Prompt(PromptBase):
 class UseCase(PromptBase):
     __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)
     id = Column(
@@ -206,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,     # 不能为空
@@ -220,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/prompt_craft/async_.py

@@ -159,7 +159,6 @@ 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)
@@ -169,8 +168,9 @@ class AsyncIntel():
             print('Use BianXieAdapter')
             self.llm = BianXieAdapter()
             
-    async def create_specific_database(self,engine, tables_to_create_names: list[str]):
-        async with engine.begin() as conn:
+    async def create_specific_database(self):
+        tables_to_create_names = ["ai_prompts","ai_usecase"]
+        async with self.engine.begin() as conn:
             # 从 metadata 中获取对应的 Table 对象
             specific_database_objects = []
             for table_name in tables_to_create_names:
@@ -254,6 +254,7 @@ class AsyncIntel():
                     )
                     records_to_sync = source_results.scalars().all()
                     if not records_to_sync:
+                        print("没有更多记录了")
                         break # 没有更多记录了
 
                     #2 max_timestamp_in_batch = datetime(1970, 1, 1) # 初始化为最早时间
@@ -912,6 +913,7 @@ class AsyncIntel():
                     prompt_id: str,
                     ExtraFormats: list[object] = [],
                     version: str = None,
+                    MIN_SUCCESS_RATE = 80.0
                     ):
         
         async with create_async_session(self.engine) as session:
@@ -974,22 +976,10 @@ class AsyncIntel():
                     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
-
-
 
+            if success_rate >= MIN_SUCCESS_RATE:
+                return "通过", success_rate, total_assertions, json.dumps(bad_case,ensure_ascii=False),
+            else:
+                return "未通过",success_rate, total_assertions, json.dumps(bad_case,ensure_ascii=False),
 
 # 整体测试d, 测试未通过d, 大模型调整再测试, 依旧不通过, 大模型裂变, 仍不通过, 互换人力

pro_craft/prompt_craft/sync.py

@@ -73,6 +73,7 @@ class Intel():
                 ):
         database_url = database_url or os.getenv("database_url")
         assert database_url
+        assert "aiomysql" not in database_url
         self.engine = create_engine(database_url, echo=False, # echo=True 仍然会打印所有执行的 SQL 语句
                                     pool_size=10,        # 连接池中保持的连接数
                                     max_overflow=20,     # 当pool_size不够时，允许临时创建的额外连接数

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

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

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

@@ -1,19 +1,19 @@
 pro_craft/__init__.py,sha256=KU9yexREoJVO5eyDNnOCzdYC8Bfo6p4Z45KWZTeO63U,698
-pro_craft/database.py,sha256=FJmknolcXpx6sYHuiS1LgXX7FV8drSpoN3RwVvRR2FI,8741
+pro_craft/database.py,sha256=NnJhtyaWbPk-KMMEPo5mRaIfyYYpxcBnt74gIzXNO_U,9302
 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=j_Jy-TNhfFehMT80kOCLelaICU-Ns3OOrPEn4s18Z8k,43846
+pro_craft/prompt_craft/async_.py,sha256=vbrOWcWohCmyAsFTBQul1tn_7Fgt6OPoWa2ccRSEQVU,43635
 pro_craft/prompt_craft/new.py,sha256=ULjGGl95vmHrOs7XECJGlaqj1NE9BypE5WnFYhGugRY,25903
-pro_craft/prompt_craft/sync.py,sha256=w-zXC9_yVxgvnotimcs2POEXaHWD6ibspdPwWL9zqXk,26178
+pro_craft/prompt_craft/sync.py,sha256=OmRVieHojyBwRZDAi8zUGBxp3-yP3sf26DDd3bWG5wM,26224
 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=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-0.1.29.dist-info/METADATA,sha256=HQm6oP2LIU-KOJLOIPZHrA23kROjkzjPFFWcKp56EUo,1689
+pro_craft-0.1.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pro_craft-0.1.29.dist-info/top_level.txt,sha256=yqYDHArnYMWpeCxkmGRwlL6sJtxiOUnYylLDx9EOgFg,10
+pro_craft-0.1.29.dist-info/RECORD,,