pro-craft 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

pro_craft/code_helper/coder.py

@@ -1,4 +1,4 @@
-from pro_craft.prompt_helper import Intel,IntellectType
+from pro_craft import Intel
 from pro_craft.log import Log
 from pro_craft.utils import extract_
 from typing import List, Dict, Any

pro_craft/file_manager.py

@@ -2,9 +2,8 @@
 from db_help.qdrant import QdrantManager
 from pydantic import BaseModel
 from datetime import datetime
-from pro_craft.utils import get_adler32_hash, embedding_inputs
-from pro_craft.prompt_helper import Intel, IntellectType
-from pro_craft.utils import extract_
+from pro_craft.utils import get_adler32_hash, embedding_inputs,extract_
+from pro_craft import Intel
 from enum import Enum
 import json
 

pro_craft/prompt_craft/async_.py

@@ -1,6 +1,6 @@
 # 测试1
 from pro_craft.utils import extract_
-from pro_craft.log import Log
+from pro_craft import logger as pro_craft_logger
 from llmada.core import BianXieAdapter, ArkAdapter
 from datetime import datetime
 from enum import Enum
@@ -68,9 +68,10 @@ class AsyncIntel():
     def __init__(self,
                  database_url = "",
                  model_name = "",
+                 logger = None,
                 ):
         database_url = database_url or os.getenv("database_url")
-        self.logger = logger or Log.logger
+        self.logger = logger or pro_craft_logger
         try:
             assert database_url
             assert 'aio' in database_url
@@ -240,7 +241,8 @@ class AsyncIntel():
         chat_history = s_prompt or system_result
         await self.save_prompt_increment_version(prompt_id,
                                 new_prompt = chat_history,
-                                input_data = " summary ",
+                                use_case = " summary ",
+                                score = 60,
                                 session = session)
         
     async def prompt_finetune_to_sql(
@@ -289,7 +291,8 @@ class AsyncIntel():
             new_prompt = prompt
         await self.save_prompt_increment_version(prompt_id = prompt_id,
                             new_prompt = new_prompt,
-                            input_data = " finetune ",
+                            use_case = " finetune ",
+                            score = 60,
                             session = session)
 
 
@@ -343,6 +346,7 @@ class AsyncIntel():
                     prompt_id = prompt_id,
                     new_prompt = "做一些处理",
                     use_case = input_,
+                    score = 60,
                     session = session
                 )
                 ai_result = await self.intellect_remove(input_data = input_data,
@@ -389,6 +393,7 @@ class AsyncIntel():
                 chat_history = input_prompt + "\nassistant:\n" + ai_result # 用聊天记录作为完整提示词
                 await self.save_prompt_increment_version(prompt_id, chat_history,
                                         use_case = input_,
+                                        score = 60,
                                         session = session)
     
             elif result_obj.action_type == "summary":
@@ -507,6 +512,7 @@ class AsyncIntel():
                 chat_history = input_prompt + "\nassistant:\n" + ai_result # 用聊天记录作为完整提示词
                 await self.save_prompt_increment_version(prompt_id, chat_history,
                                         use_case = input_,
+                                        score = 60,
                                         session = session)
     
             elif result_obj.action_type == "summary":

pro_craft/prompt_craft/new.py

@@ -1,7 +1,7 @@
 # 测试1
 
 from pro_craft.utils import extract_
-from pro_craft.log import Log
+from pro_craft import logger as pro_craft_logger
 from pro_craft.database import Prompt, UseCase, PromptBase
 from pro_craft.utils import create_session, create_async_session
 from llmada.core import BianXieAdapter, ArkAdapter
@@ -88,7 +88,7 @@ class IntelNew():
         else:
             print('Use BianXieAdapter')
             self.llm = BianXieAdapter()
-        self.logger = logger or Log.logger
+        self.logger = logger or pro_craft_logger
         
     def _get_latest_prompt_version(self,target_prompt_id,session):
         """

pro_craft/prompt_craft/sync.py

@@ -1,7 +1,7 @@
 # 测试1
 
 from pro_craft.utils import extract_
-from pro_craft.log import Log
+from pro_craft import logger as pro_craft_logger
 from pro_craft.database import Prompt, UseCase, PromptBase
 from pro_craft.utils import create_session, create_async_session
 from llmada.core import BianXieAdapter, ArkAdapter
@@ -88,7 +88,7 @@ class Intel():
         else:
             print('Use BianXieAdapter')
             self.llm = BianXieAdapter()
-        self.logger = logger or Log.logger
+        self.logger = logger or pro_craft_logger
         
     def _get_latest_prompt_version(self,target_prompt_id,session):
         """
@@ -397,14 +397,10 @@ class Intel():
                     prompt_id = prompt_id,
                     new_prompt = "做一些处理",
                     use_case = input_,
+                    score = 60,
                     session = session
                 )
-                ai_result = self.intellect_remove(input_data = input_data,
-                                       output_format = output_format,
-                                       prompt_id = prompt_id,
-                                       version = version,
-                                       inference_save_case = inference_save_case
-                                       )
+                ai_result = "初始化完成"
                 return ai_result
             prompt = result_obj.prompt
             if result_obj.action_type == "inference":

pro_craft/server/mcp/{weather.py → prompt.py}

@@ -1,7 +1,6 @@
-from datetime import datetime
 from mcp.server.fastmcp import FastMCP
 
-from pro_craft.prompt_helper import IntellectType, Intel
+from pro_craft import Intel
 
 def create_mcp(database_url: str,
                   slave_database_url: str,

pro_craft/server/router/prompt.py

@@ -1,9 +1,8 @@
 
 
 from fastapi import APIRouter
-from pro_craft.prompt_helper import IntellectType,Intel
+from pro_craft import Intel
 from pro_craft.utils import create_session
-import os
 
 def create_router(database_url: str,
                   slave_database_url: str,
@@ -30,7 +29,8 @@ def create_router(database_url: str,
         tags=["prompt"] # 这里使用 Depends 确保每次请求都验证
     )
 
-    @router.get("/push_order")
+    @router.get("/push_order",
+                description="可选 train,inference,summary,finetune,patch",)
     async def push_order(demand: str, prompt_id: str, action_type: str = "train"):
         result = intels.push_action_order(
             demand=demand,
@@ -53,4 +53,25 @@ def create_router(database_url: str,
         result = intels.sync_prompt_data_to_database(slave_database_url)
         return {"message": "success","result":result}
     
+
+    @router.get("/roll_back")
+    async def roll_back(prompt_id:str,version:str):
+        with create_session(intels.engine) as session:
+            result = intels.get_prompts_from_sql(
+                prompt_id=prompt_id,
+                version = version,
+                session=session
+            )
+            assert result.version == version
+            intels.save_prompt_increment_version(
+                            prompt_id = prompt_id,
+                            new_prompt = result.prompt,
+                            use_case = "roll_back",
+                            action_type = "inference",
+                            demand = "",
+                            score = 61,
+                            session = session)
+        return {"message": "success","result":result.version}
+    
     return router
+

{pro_craft-0.1.15.dist-info → pro_craft-0.1.17.dist-info}/METADATA

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

pro_craft-0.1.17.dist-info/RECORD

@@ -0,0 +1,20 @@
+pro_craft/__init__.py,sha256=KU9yexREoJVO5eyDNnOCzdYC8Bfo6p4Z45KWZTeO63U,698
+pro_craft/database.py,sha256=5dV-h9dVaS6euHLCtf0gYfq2pchl2QFdb2PEM4gTEU4,8740
+pro_craft/file_manager.py,sha256=93m3AkS6R88pJ5NBwaePyngM-YwSuiPh-9Sx7YPh3B8,3748
+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=iX1Ok2SrfU2jIR9ScgDoXdLN43lsVSdLpGArP_Hg3wc,27519
+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=uuVJ3eT3Ukavh0kX_v6GXXeZEd7JhlXiEUpqtLHCfXE,25454
+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=qbwZPL-F_zFwBo9ua46aYyfgShlxfxW5VcoHn5kgYis,2765
+pro_craft-0.1.17.dist-info/METADATA,sha256=9X9WeFYQzgbYCwVd9GSuilC1w46MTBZl_34HUR_QLXs,1800
+pro_craft-0.1.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pro_craft-0.1.17.dist-info/top_level.txt,sha256=yqYDHArnYMWpeCxkmGRwlL6sJtxiOUnYylLDx9EOgFg,10
+pro_craft-0.1.17.dist-info/RECORD,,

pro_craft-0.1.15.dist-info/RECORD

@@ -1,20 +0,0 @@
-pro_craft/__init__.py,sha256=KU9yexREoJVO5eyDNnOCzdYC8Bfo6p4Z45KWZTeO63U,698
-pro_craft/database.py,sha256=5dV-h9dVaS6euHLCtf0gYfq2pchl2QFdb2PEM4gTEU4,8740
-pro_craft/file_manager.py,sha256=2j7lCt9L4mtvAy8_76ibTthXLwKKmVatWIB3DSvQM7U,3805
-pro_craft/log.py,sha256=x9RS_0LITN2SE8dcVaEcUFdcfr__jCYpFkIkVmqE5f0,3061
-pro_craft/utils.py,sha256=R1DFkS4dsm5dIhg8lLTgBBvItvIYyyojROdh-ykqiYk,5250
-pro_craft/code_helper/coder.py,sha256=NXglF1KiPtGe4HZN0MZvFJ8p9Iyd5kzIt72DQGgRwXA,24715
-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=EVCehdVPRdiY412DxeWyToKAe27lamUbMkbW6gbykFs,27246
-pro_craft/prompt_craft/evals.py,sha256=XzaaQgA-Vgo8MNPYTdN0hyFJNmgTvw2jroWmeA02pBs,1847
-pro_craft/prompt_craft/new.py,sha256=W62RYG2XH6wyC45BKKsPcDP6_QL2O6_B94FvyHt5jrE,25869
-pro_craft/prompt_craft/sync.py,sha256=bHVB5XEAmTTVK9QsUaBNlLQiq6GHyu0FT1HZeXWFC_w,25738
-pro_craft/server/mcp/__init__.py,sha256=4dbl-lFcm0r2tkOP04OxqiZG2jR-rqF181qi2AfU6UA,123
-pro_craft/server/mcp/weather.py,sha256=46mwcRS-6DFHyOSirgphkpb0Bmtt8Dhx-xHjitQ0FWo,1304
-pro_craft/server/router/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pro_craft/server/router/prompt.py,sha256=VbxuvXvxJ-wLEmMnk7ec3bsIF-WB_JGh7GIYxZkXyWc,1914
-pro_craft-0.1.15.dist-info/METADATA,sha256=ZAtsZ0wyqhD6HHmHxZzXd7deDquPvK6V_bHF0vr5JKk,1800
-pro_craft-0.1.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pro_craft-0.1.15.dist-info/top_level.txt,sha256=yqYDHArnYMWpeCxkmGRwlL6sJtxiOUnYylLDx9EOgFg,10
-pro_craft-0.1.15.dist-info/RECORD,,