pro-craft 0.1.21__tar.gz → 0.1.23__tar.gz

{pro_craft-0.1.21 → pro_craft-0.1.23}/PKG-INFO

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

{pro_craft-0.1.21 → pro_craft-0.1.23}/pyproject.toml

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

{pro_craft-0.1.21 → pro_craft-0.1.23}/src/pro_craft/prompt_craft/async_.py

@@ -23,6 +23,9 @@ from json.decoder import JSONDecodeError
 class IntellectRemoveFormatError(Exception):
     pass
 
+class IntellectRemoveError(Exception):
+    pass
+
 
 def slog(s, target: str = "target",logger = None):
     COLOR_GREEN = "\033[92m"
@@ -310,8 +313,8 @@ class AsyncIntel():
         async with create_async_session(self.engine) as session:
 
             latest_prompt = await self.get_prompts_from_sql(prompt_id=prompt_id,session=session)
-
-            await self.save_prompt_increment_version(prompt_id=latest_prompt.prompt_id,
+            if latest_prompt:
+                await self.save_prompt_increment_version(prompt_id=latest_prompt.prompt_id,
                                     new_prompt = latest_prompt.prompt,
                                     use_case = latest_prompt.use_case,
                                     action_type=action_type,
@@ -319,8 +322,21 @@ class AsyncIntel():
                                     score=latest_prompt.score,
                                     session=session
                                     )
+                return "success"
+            else:
+                await self.save_prompt_increment_version(prompt_id=prompt_id,
+                                    new_prompt = demand,
+                                    use_case = "init",
+                                    action_type="inference",
+                                    demand=demand,
+                                    score=60,
+                                    session=session
+                                    )
+                return "init"
+
+
+            
             
-            return "success"
 
 
 
@@ -330,7 +346,6 @@ class AsyncIntel():
                     prompt_id: str,
                     version: str = None,
                     inference_save_case = True,
-                    push_patch = False,
                     ):
         if isinstance(input_data,dict):
             input_ = json.dumps(input_data,ensure_ascii=False)
@@ -340,18 +355,8 @@ class AsyncIntel():
         # 查数据库, 获取最新提示词对象
         async with create_async_session(self.engine) as session:
             result_obj = await self.get_prompts_from_sql(prompt_id=prompt_id,session=session)
-
-
             if result_obj is None:
-                await self.save_prompt_increment_version(
-                    prompt_id = prompt_id,
-                    new_prompt = "做一些处理",
-                    use_case = input_,
-                    score = 60,
-                    session = session
-                )
-                ai_result = "初始化完成"
-                return ai_result
+                raise IntellectRemoveError("不存在的prompt_id")
 
             prompt = result_obj.prompt
             if result_obj.action_type == "inference":
@@ -427,18 +432,15 @@ class AsyncIntel():
                                             )
                 ai_result = await self.llm.aproduct(prompt + output_format + "\n-----input----\n" +  input_)
             elif result_obj.action_type == "patch":
-                
                 demand = result_obj.demand
                 assert demand
-
                 chat_history = prompt + demand
                 ai_result = await self.llm.aproduct(chat_history + output_format + "\n-----input----\n" +  input_)
-
-                if push_patch:
-                    self.save_prompt_increment_version(prompt_id, chat_history,
-                                            use_case = input_,
-                                            score = 60,
-                                            session = session)
+                self.save_prompt_increment_version(prompt_id, 
+                                                   chat_history,
+                                                    use_case = input_,
+                                                    score = 60,
+                                                    session = session)
 
             else:
                 raise
@@ -583,7 +585,6 @@ class AsyncIntel():
                     ExtraFormats: list[object] = [],
                     version: str = None,
                     inference_save_case = True,
-                    push_patch = False,
                     ):
                 
         base_format_prompt = """
@@ -601,7 +602,6 @@ class AsyncIntel():
                     prompt_id=prompt_id,
                     version=version,
                     inference_save_case=inference_save_case,
-                    push_patch = push_patch,
                 )
 
         try:
@@ -631,7 +631,6 @@ class AsyncIntel():
                     ExtraFormats: list[object] = [],
                     version: str = None,
                     inference_save_case = True,
-                    push_patch = False,
                     ):
                 
         async with create_async_session(self.engine) as session:
@@ -649,7 +648,6 @@ class AsyncIntel():
                     ExtraFormats = ExtraFormats,
                     version = version,
                     inference_save_case = inference_save_case,
-                    push_patch = push_patch
                 )
             )
         results = await asyncio.gather(*tasks, return_exceptions=False)

{pro_craft-0.1.21 → pro_craft-0.1.23}/src/pro_craft.egg-info/PKG-INFO

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