pro-craft 0.1.32__tar.gz → 0.1.34__tar.gz

{pro_craft-0.1.32 → pro_craft-0.1.34}/PKG-INFO

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

{pro_craft-0.1.32 → pro_craft-0.1.34}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pro-craft"
-version = "0.1.32"
+version = "0.1.34"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"

{pro_craft-0.1.32 → pro_craft-0.1.34}/src/pro_craft/database.py

@@ -235,7 +235,7 @@ class UseCase(PromptBase):
 
     faired_time = Column(
         Integer,
-        nullable=True,     # 不能为空
+        nullable=False,     # 不能为空
         comment="失败次数"
     )
 

{pro_craft-0.1.32 → pro_craft-0.1.34}/src/pro_craft/prompt_craft/async_ copy.py

@@ -162,11 +162,10 @@ class AsyncIntel():
 
         if model_name in ["gemini-2.5-flash-preview-05-20-nothinking",]:
             self.llm = BianXieAdapter(model_name = model_name)
-        elif model_name in ["doubao-1-5-pro-256k-250115",]:
+        elif model_name in ["doubao-1-5-pro-256k-250115","doubao-1-5-pro-32k-250115"]:
             self.llm = ArkAdapter(model_name = model_name)
         else:
-            print('Use BianXieAdapter')
-            self.llm = BianXieAdapter()
+            raise Exception("error llm name")
             
     async def create_specific_database(self):
         tables_to_create_names = ["ai_prompts","ai_usecase"]
@@ -400,20 +399,30 @@ class AsyncIntel():
     async def save_use_case_by_sql(self,
                              prompt_id: str,
                              use_case:str = "",
+                             timestamp = "",
                              output = "",
                              solution: str = "",
+                             faired_time  = 0,
                              session = None
                             ):
+
         """
         从sql保存提示词
         """
         #TODO 存之前保证数据库中相同的prompt_id中没有重复的use_case
+        use_cases = await self.get_use_case_by_sql(target_prompt_id = prompt_id,
+                                  session = session)
+        for use_case_old in use_cases:
+            if use_case == use_case_old.use_case:
+                print("用例已经存在")
+                return 
         
-        use_case = UseCase(prompt_id=prompt_id, 
+        use_case = UseCase(prompt_id=prompt_id,
                         use_case = use_case,
+                        timestamp = timestamp,
                         output = output,
                         solution = solution,
-                        faired_time = 1,
+                        faired_time = faired_time,
                         )
 
         session.add(use_case)
@@ -559,8 +568,10 @@ class AsyncIntel():
                 if inference_save_case:
                     await self.save_use_case_by_sql(prompt_id,
                                         use_case = input_,
+                                        timestamp = datetime.now(),
                                         output = ai_result,
                                         solution = "备注/理想回复",
+                                        faired_time = 0,
                                         session = session,
                                         )
                     
@@ -651,6 +662,7 @@ class AsyncIntel():
                                             session = session
                                             )
                 ai_result = await self.llm.aproduct(prompt + output_format + "\nuser:" +  input_)
+
             elif result_obj.action_type == "patch":
                 demand = result_obj.demand
                 assert demand
@@ -713,8 +725,10 @@ class AsyncIntel():
                 if inference_save_case:
                     await self.save_use_case_by_sql(prompt_id,
                                         use_case = input_,
+                                        timestamp = datetime.now(),
                                         output = ai_result,
                                         solution = "备注/理想回复",
+                                        faired_time = 0,
                                         session = session,
                                         )
                     

{pro_craft-0.1.32 → pro_craft-0.1.34}/src/pro_craft/prompt_craft/async_.py

@@ -162,11 +162,10 @@ class AsyncIntel():
 
         if model_name in ["gemini-2.5-flash-preview-05-20-nothinking",]:
             self.llm = BianXieAdapter(model_name = model_name)
-        elif model_name in ["doubao-1-5-pro-256k-250115",]:
+        elif model_name in ["doubao-1-5-pro-256k-250115","doubao-1-5-pro-32k-250115"]:
             self.llm = ArkAdapter(model_name = model_name)
         else:
-            print('Use BianXieAdapter')
-            self.llm = BianXieAdapter()
+            raise Exception("error llm name")
             
     async def create_specific_database(self):
         tables_to_create_names = ["ai_prompts","ai_usecase"]
@@ -400,19 +399,30 @@ class AsyncIntel():
     async def save_use_case_by_sql(self,
                              prompt_id: str,
                              use_case:str = "",
+                             timestamp = "",
                              output = "",
                              solution: str = "",
+                             faired_time  = 0,
                              session = None
                             ):
+
         """
         从sql保存提示词
         """
         #TODO 存之前保证数据库中相同的prompt_id中没有重复的use_case
-
-        use_case = UseCase(prompt_id=prompt_id, 
+        use_cases = await self.get_use_case_by_sql(target_prompt_id = prompt_id,
+                                  session = session)
+        for use_case_old in use_cases:
+            if use_case == use_case_old.use_case:
+                print("用例已经存在")
+                return 
+        
+        use_case = UseCase(prompt_id=prompt_id,
                         use_case = use_case,
+                        timestamp = timestamp,
                         output = output,
                         solution = solution,
+                        faired_time = faired_time,
                         )
 
         session.add(use_case)
@@ -558,8 +568,10 @@ class AsyncIntel():
                 if inference_save_case:
                     await self.save_use_case_by_sql(prompt_id,
                                         use_case = input_,
+                                        timestamp = datetime.now(),
                                         output = ai_result,
                                         solution = "备注/理想回复",
+                                        faired_time = 0,
                                         session = session,
                                         )
                     
@@ -650,6 +662,7 @@ class AsyncIntel():
                                             session = session
                                             )
                 ai_result = await self.llm.aproduct(prompt + output_format + "\nuser:" +  input_)
+
             elif result_obj.action_type == "patch":
                 demand = result_obj.demand
                 assert demand
@@ -712,8 +725,10 @@ class AsyncIntel():
                 if inference_save_case:
                     await self.save_use_case_by_sql(prompt_id,
                                         use_case = input_,
+                                        timestamp = datetime.now(),
                                         output = ai_result,
                                         solution = "备注/理想回复",
+                                        faired_time = 0,
                                         session = session,
                                         )
                     

{pro_craft-0.1.32 → pro_craft-0.1.34}/src/pro_craft/prompt_craft/sync.py

@@ -86,11 +86,10 @@ class Intel():
         
         if model_name in ["gemini-2.5-flash-preview-05-20-nothinking",]:
             self.llm = BianXieAdapter(model_name = model_name)
-        elif model_name in ["doubao-1-5-pro-256k-250115",]:
+        elif model_name in ["doubao-1-5-pro-256k-250115","doubao-1-5-pro-32k-250115"]:
             self.llm = ArkAdapter(model_name = model_name)
         else:
-            print('Use BianXieAdapter')
-            self.llm = BianXieAdapter()
+            raise Exception("error llm name")
         self.logger = logger or pro_craft_logger
         
     def _get_latest_prompt_version(self,target_prompt_id,session):

{pro_craft-0.1.32 → pro_craft-0.1.34}/src/pro_craft.egg-info/PKG-INFO

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