pro-craft 0.1.37__tar.gz → 0.1.39__tar.gz

{pro_craft-0.1.37 → pro_craft-0.1.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pro-craft
-Version: 0.1.37
+Version: 0.1.39
 Summary: Add your description here
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -9,6 +9,7 @@ Requires-Dist: anyio>=4.11.0
 Requires-Dist: db-help>=0.2.2
 Requires-Dist: fastapi>=0.119.0
 Requires-Dist: llmada>=1.1.11
+Requires-Dist: mcp[cli]>=1.19.0
 Requires-Dist: plotly>=6.3.1
 Requires-Dist: pyyaml>=6.0.3
 Requires-Dist: toml>=0.10.2

{pro_craft-0.1.37 → pro_craft-0.1.39}/pyproject.toml

@@ -1,10 +1,10 @@
 [project]
 name = "pro-craft"
-version = "0.1.37"
+version = "0.1.39"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"
-dependencies = [ "aiomysql>=0.2.0", "anyio>=4.11.0", "db-help>=0.2.2", "fastapi>=0.119.0", "llmada>=1.1.11", "plotly>=6.3.1", "pyyaml>=6.0.3", "toml>=0.10.2", "utils-tool==0.1.3", "uvicorn>=0.38.0",]
+dependencies = [ "aiomysql>=0.2.0", "anyio>=4.11.0", "db-help>=0.2.2", "fastapi>=0.119.0", "llmada>=1.1.11", "mcp[cli]>=1.19.0", "plotly>=6.3.1", "pyyaml>=6.0.3", "toml>=0.10.2", "utils-tool==0.1.3", "uvicorn>=0.38.0",]
 
 [tool.setuptools.package-data]
 pro_craft = [ "config.yaml",]

{pro_craft-0.1.37 → pro_craft-0.1.39}/src/pro_craft/database.py

@@ -1,5 +1,5 @@
 
-from sqlalchemy import Column, Integer, String, Text, DateTime, text, UniqueConstraint, Boolean, func
+from sqlalchemy import Column, Integer, String, Text, DateTime, text, UniqueConstraint, Boolean, func, Float
 from sqlalchemy.orm import declarative_base
 
 from datetime import datetime, timedelta
@@ -248,3 +248,72 @@ class UseCase(PromptBase):
                 f"is_deleted='{self.is_deleted}...'>")
 
 
+class EvalsInfo(PromptBase):
+    __tablename__ = 'ai_evals' # 数据库中的表名，你可以改成你希望的名字
+
+    # __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(
+        Integer,
+        primary_key=True,
+        autoincrement=True, # 自动递增
+        nullable=False,     # 不能为空
+        comment="Primary key ID"
+    )
+
+    # prompt_id (varchar 255, not null, unique)
+    # 你的属性表中 prompt_id 为 varchar, 255 (length), true (not null)
+    prompt_id = Column(
+        String(255),        # VARCHAR 类型，长度 255
+        nullable=False,     # 不能为空    # 必须是唯一的，这会创建唯一索引
+        comment="Unique identifier for the prompt"
+    )
+    status = Column(
+        String(255),        # VARCHAR 类型，长度 255
+        nullable=False,     # 不能为空    # 必须是唯一的，这会创建唯一索引
+        comment="通过/未通过"
+    )
+
+    score = Column(
+        Float,
+        nullable=False,     # 不能为空
+        comment="失败次数"
+    )
+
+    total = Column(
+        Integer,
+        nullable=False,     # 不能为空
+        comment="失败次数"
+    )
+
+    # prompt (text, not null)
+    # 你的属性表中 prompt 为 text, true (not null)
+    bad_case = Column(
+        Text,               # TEXT 类型，适用于长文本
+        nullable=False,     # 不能为空
+        comment="用例"
+    )
+
+    timestamp = Column(
+        DateTime,
+        nullable=False,      # 不能为空
+        server_default=text('CURRENT_TIMESTAMP'),
+        onupdate=text('CURRENT_TIMESTAMP'),
+        comment="时间戳"
+    )
+
+
+    is_deleted = Column(Boolean, default=False, server_default=text('0')) 
+
+    # 定义 __repr__ 方法以便打印对象时有清晰的表示
+    def __repr__(self):
+        return (f"<Prompt(id={self.id}, prompt_id='{self.prompt_id}', "
+                f"status='{self.status}...', score='{self.score}')>"
+                f"total='{self.total}...', score='{self.score}')>"
+                f"is_deleted='{self.is_deleted}...'>")
+

{pro_craft-0.1.37 → pro_craft-0.1.39}/src/pro_craft/prompt_craft/async_.py

@@ -25,14 +25,14 @@ from datetime import datetime, timedelta
 from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine, async_sessionmaker
 from sqlalchemy import select, and_ # 引入 select 和 and_
 from sqlalchemy.orm import class_mapper # 用于检查对象是否是持久化的
-import tqdm
+
 from tqdm.asyncio import tqdm
 import pandas as pd
 import plotly.graph_objects as go
 from pro_craft import super_log
 
 BATCH_SIZE = int(os.getenv("DATABASE_SYNC_BATCH_SIZE",100))
-
+print(12)
 
 def fix_broken_json_string(broken_json_str):
     # 移除 BOM
@@ -421,6 +421,10 @@ class AsyncIntel():
                     inference_save_case = True,
                     change_case = False,
                     ):
+        """
+        自定自动化执行命令的方法, 
+        不涉及严格的校验, 主要职能在自动化的修改提示词, 或者管理提示词上
+        """
         if isinstance(input_data,dict):
             input_ = json.dumps(input_data,ensure_ascii=False)
         elif isinstance(input_data,str):
@@ -634,7 +638,16 @@ class AsyncIntel():
                     ExtraFormats: list[object] = [],
                     version: str = None,
                     inference_save_case = True,
+                    ConTent_Function = None,
+                    AConTent_Function = None,
                     ):
+        """
+        这个format 是严格校验模式, 是interllect 的增强版, 会主动校验内容,并及时抛出异常(或者伺机修正)
+        ConTent_Function
+        AConTent_Function
+        两种方式的传入方式, 内容未通过就抛出异常
+
+        """
                 
         base_format_prompt = """
 按照一定格式输出, 以便可以通过如下校验
@@ -642,7 +655,7 @@ class AsyncIntel():
 使用以下正则检出
 "```json([\s\S]*?)```"
 使用以下方式验证
-"""
+"""     
         if OutputFormat:
             output_format = base_format_prompt + "\n".join([inspect.getsource(outputformat) for outputformat in ExtraFormats]) + inspect.getsource(OutputFormat)
         else:
@@ -676,6 +689,12 @@ class AsyncIntel():
 
             except Exception as e:
                 raise Exception(f"Error {prompt_id} : {e}") from e
+            
+        if ConTent_Function:
+            ConTent_Function(ai_result)
+        
+        if AConTent_Function:
+            await AConTent_Function(ai_result)
 
         return ai_result
     
@@ -686,6 +705,7 @@ class AsyncIntel():
                     ExtraFormats: list[object] = [],
                     version: str = None,
                     inference_save_case = True,
+                    **kwargs,
                     ):
                 
         async with create_async_session(self.engine) as session:
@@ -705,6 +725,7 @@ class AsyncIntel():
                     ExtraFormats = ExtraFormats,
                     version = version,
                     inference_save_case = inference_save_case,
+                    **kwargs,
                 )
             )
         results = await asyncio.gather(*tasks, return_exceptions=False)
@@ -717,8 +738,7 @@ class AsyncIntel():
                 # 修改逻辑
                 assert kwargs.get('input_data') # 要求一定要有data入参
                 input_data = kwargs.get('input_data')
-                OutputFormat = kwargs.get('OutputFormat','')
-
+                kwargs.pop(input_data)
                 if isinstance(input_data,dict):
                     input_ = output_ = json.dumps(input_data,ensure_ascii=False)
                 elif isinstance(input_data,str):
@@ -727,7 +747,8 @@ class AsyncIntel():
                 output_ = await self.intellect_format(
                         input_data = input_data,
                         prompt_id = prompt_id,
-                        OutputFormat = OutputFormat,
+                        **kwargs,
+                        
                 )
 
                 #######
@@ -740,10 +761,12 @@ class AsyncIntel():
     async def intellect_format_eval(self,
                     OutputFormat: object,
                     prompt_id: str,
+                    database_url = None,
                     ExtraFormats: list[object] = [],
                     version: str = None,
                     MIN_SUCCESS_RATE = 80.0,
                     ConTent_Function = None,
+                    AConTent_Function = None,
                     ):
         """
         ConTent_Function: 
@@ -751,20 +774,32 @@ class AsyncIntel():
         # TODO llm 评价 eval
         """
         async with create_async_session(self.engine) as session:
-            use_cases = await self.get_use_case(target_prompt_id=prompt_id,session=session)
             prompt_result = await self.get_prompt_safe(prompt_id=prompt_id,
                                                                    session=session)
             if prompt_result is None:
                 raise IntellectRemoveError("不存在的prompt_id")
             if prompt_result.action_type != "inference":
                 raise IntellectRemoveError("请在inference模式下使用次类")
-            
+
+        if database_url:
+            eval_engine = create_async_engine(database_url, echo=False,
+                                pool_size=10,        # 连接池中保持的连接数
+                                max_overflow=20,     # 当pool_size不够时，允许临时创建的额外连接数
+                                pool_recycle=3600,   # 每小时回收一次连接
+                                pool_pre_ping=True,  # 使用前检查连接活性
+                                pool_timeout=30      # 等待连接池中连接的最长时间（秒）
+                                        )
+        else:
+            eval_engine = self.engine
+        async with create_async_session(eval_engine) as eval_session:
+            use_cases = await self.get_use_case(target_prompt_id=prompt_id,session=eval_session)
 
             total_assertions = len(use_cases)
             result_cases = []
 
             async def evals_func(use_case,prompt_id,OutputFormat,ExtraFormats,version):
                 try:
+
                     # 这里将参数传入
                     ai_result = await self.intellect_format(
                         input_data = use_case.use_case,
@@ -773,18 +808,24 @@ class AsyncIntel():
                         ExtraFormats = ExtraFormats,
                         version = version,
                         inference_save_case = False,
+                        ConTent_Function = ConTent_Function,
+                        AConTent_Function = AConTent_Function,
                     )
-                    if ConTent_Function:
-                        ConTent_Function()
+
                     result_cases.append({"type":"Successful","case":use_case.use_case,"reply":f"pass"})
-                    use_case.output = ai_result
+                    use_case.output = json.dumps(ai_result,ensure_ascii=False,indent=4)
+
+
                 except IntellectRemoveFormatError as e:
                     result_cases.append({"type":"FAILED","case":use_case.use_case,"reply":f"{e}"})
                     use_case.output = f"{"FAILED"}-{e}"
+                    use_case.faired_time +=1
+
                 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()
+                    use_case.faired_time +=1
+
 
             tasks = []
             for use_case in use_cases:
@@ -800,6 +841,7 @@ class AsyncIntel():
             await tqdm.gather(*tasks,total=len(tasks))
             # await asyncio.gather(*tasks, return_exceptions=False)
 
+            await eval_session.commit()
 
             successful_assertions = 0
             bad_case = []
@@ -811,14 +853,23 @@ class AsyncIntel():
 
             success_rate = (successful_assertions / total_assertions) * 100
 
+
             if success_rate >= MIN_SUCCESS_RATE:
+                self.eval_df.loc[len(self.eval_df)] = {"name":prompt_id,
+                                                   'status':"通过",
+                                                   "score":success_rate,
+                                                   "total":str(total_assertions),
+                                                   "bad_case":json.dumps(bad_case,ensure_ascii=False)}
                 return "通过", success_rate, str(total_assertions), json.dumps(bad_case,ensure_ascii=False),
             else:
+                self.eval_df.loc[len(self.eval_df)] = {"name":prompt_id,
+                                                   'status':"未通过",
+                                                   "score":success_rate,
+                                                   "total":str(total_assertions),
+                                                   "bad_case":json.dumps(bad_case,ensure_ascii=False)}
                 return "未通过",success_rate, str(total_assertions), json.dumps(bad_case,ensure_ascii=False),
 
-
-
-    def draw_data(self):
+    def draw_data(self,save_html_path):
         df = self.eval_df
         # --- 可视化部分 ---
         fig = go.Figure()
@@ -899,19 +950,8 @@ class AsyncIntel():
         )
 
         fig.show()
-        pass
+        if save_html_path:
+            fig.write_html(save_html_path)
 
-    async def _evals(self,prompt_id, OutputFormat, ExtraFormats_list = [],**kwargs):
-        
-        status,score, total, bad_case = await self.intellect_format_eval(
-            prompt_id=prompt_id,
-            OutputFormat = OutputFormat,
-            ExtraFormats = ExtraFormats_list,
-            version = None,
-            **kwargs
-        )
-        self.df.loc[len(self.df)] = {"name":prompt_id,
-                           'status':status,"score":score,
-                           "total":total,"bad_case":bad_case}
 
 # 整体测试d, 测试未通过d, 大模型调整再测试, 依旧不通过, 大模型裂变, 仍不通过, 互换人力

pro_craft-0.1.39/src/pro_craft/server/mcp/models.py

@@ -0,0 +1,35 @@
+from typing import Dict, Any, Optional, List
+from pydantic import BaseModel, Field, model_validator, field_validator, RootModel
+import re
+
+class PushOrderRequest(BaseModel):
+    demand: str = Field(None, description="信息")
+    prompt_id: str = Field(..., description="提示词id")
+    action_type: str = Field(..., description="执行动作",min_length=1, max_length=10)
+
+    @field_validator('action_type')
+    @classmethod
+    def validate_action_type(cls, v: str) -> str:
+        if v in ['train','inference','summary','finetune','patch']:
+            return v
+        else:
+            raise ValueError(f"无效action_type: {v}")
+
+class GetPromptRequest(BaseModel):
+    prompt_id: str = Field(..., description="提示词id")
+
+class UpdatePromptRequest(BaseModel):
+    prompt_id: str = Field(..., description="提示词id")
+    prompt: str = Field(..., description="新的提示词")
+
+class RollBackPromptRequest(BaseModel):
+    prompt_id: str = Field(..., description="提示词id")
+    version: str = Field(..., description="版本号")
+
+class SyncDataBaseRequest(BaseModel):
+    slave_database_url: str = Field(None, description="从属数据库url")
+    
+
+class PromptResponse(BaseModel):
+    msg: str = Field(..., description="信息")
+    content: str = None

pro_craft-0.1.39/src/pro_craft/server/mcp/prompt.py

@@ -0,0 +1,78 @@
+from mcp.server.fastmcp import FastMCP
+
+from pro_craft import Intel,AsyncIntel
+from .models import *
+
+def create_mcp(database_url: str,
+                  slave_database_url: str,
+                  model_name: str,
+                  logger = None):
+    # region MCP Weather
+    mcp = FastMCP("Prompt")
+
+    intels = AsyncIntel(
+        database_url=database_url,
+        model_name=model_name,
+        logger=logger
+        )
+
+    @mcp.tool()
+    async def push_order(demand: str, prompt_id: str, action_type: str):
+        """
+        希望大模型进行哪种模式的调整
+        demand: str = Field(None, description="信息")
+        prompt_id: str = Field(..., description="提示词id")
+        action_type: str = Field(..., description="执行动作",min_length=1, max_length=10)
+        """
+        try:
+            PushOrderRequest(demand=demand,prompt_id=prompt_id,action_type=action_type)
+            result = await intels.push_action_order(
+                demand=demand,
+                prompt_id=prompt_id,
+                action_type=action_type
+            )
+            return PromptResponse(msg = "success",content=result)
+        except Exception as e:
+            return f"Error : {e}"
+        
+    @mcp.tool()
+    async def get_registered_prompt():
+        "获取以注册的可修改的提示词id"
+        try:
+            result = ["memorycard-format",
+            "memorycard-polish",
+            "memorycard-merge",
+            "memorycard-score",
+            "memorycard-generate-content",
+            "user-overview",
+            "user-relationship-extraction",
+            "avatar-brief",
+            "avatar-personality-extraction",
+            "avatar-desensitization",
+            "biograph-free-writer",
+            "biograph-paid-title",
+            "biograph-outline",
+            "biograph-brief",
+            "biograph-extract-person-name",
+            "biograph-extract-place",
+            "biograph-extract-material",
+            "biograph-writer"]
+
+            return PromptResponse(msg = "success",content=result)
+        except Exception as e:
+            return f"Error : {e}"
+
+
+    @mcp.tool()
+    async def sync_database():
+        try:
+            result = await intels.sync_production_database(slave_database_url)
+            return PromptResponse(msg = "success",content=result)
+        except Exception as e:
+            return f"Error : {e}"
+
+    return mcp
+
+if __name__ == "__main__":
+    mcp = create_mcp()
+    mcp.run(transport="streamable-http")

pro_craft-0.1.39/src/pro_craft/server/router/models.py

@@ -0,0 +1,35 @@
+from typing import Dict, Any, Optional, List
+from pydantic import BaseModel, Field, model_validator, field_validator, RootModel
+import re
+
+class PushOrderRequest(BaseModel):
+    demand: str = Field(None, description="信息")
+    prompt_id: str = Field(..., description="提示词id")
+    action_type: str = Field(..., description="执行动作",min_length=1, max_length=10)
+
+    @field_validator('action_type')
+    @classmethod
+    def validate_action_type(cls, v: str) -> str:
+        if v in ['train','inference','summary','finetune','patch']:
+            return v
+        else:
+            raise ValueError(f"无效action_type: {v}")
+
+class GetPromptRequest(BaseModel):
+    prompt_id: str = Field(..., description="提示词id")
+
+class UpdatePromptRequest(BaseModel):
+    prompt_id: str = Field(..., description="提示词id")
+    prompt: str = Field(..., description="新的提示词")
+
+class RollBackPromptRequest(BaseModel):
+    prompt_id: str = Field(..., description="提示词id")
+    version: str = Field(..., description="版本号")
+
+class SyncDataBaseRequest(BaseModel):
+    slave_database_url: str = Field(None, description="从属数据库url")
+    
+
+class PromptResponse(BaseModel):
+    msg: str = Field(..., description="信息")
+    content: str = None

pro_craft-0.1.39/src/pro_craft/server/router/prompt.py

@@ -0,0 +1,163 @@
+
+
+from fastapi import APIRouter
+from pro_craft import Intel,AsyncIntel
+from pro_craft.utils import create_async_session
+from fastapi import FastAPI, HTTPException
+from .models import *
+
+def create_router(database_url: str,
+                  slave_database_url: str,
+                  model_name: str,
+                  logger = None):
+    """
+    # TODO 整理改为异步
+    创建一个包含 ProCraft 路由的 FastAPI APIRouter 实例。
+
+    Args:
+        database_url (str): 数据库连接字符串。
+        model_name (str): 用于 Intel 实例的模型名称。
+        api_key_secret (str, optional): 用于验证 API Key 的秘密字符串。
+                                        如果提供，它将覆盖环境变量 PRO_CRAFT_API_KEY。
+                                        如果都不提供，会使用硬编码的 'your_default_secret_key'。
+    Returns:
+        APIRouter: 配置好的 FastAPI APIRouter 实例。
+    """
+
+    intels = AsyncIntel(
+        database_url=database_url,
+        model_name=model_name,
+        logger=logger
+        )
+
+    router = APIRouter(
+        tags=["prompt"] # 这里使用 Depends 确保每次请求都验证
+    )
+
+    # 自动修改
+    @router.post("/push_order",
+                description="可选 train,inference,summary,finetune,patch",
+                response_model=PromptResponse,
+                )
+    async def push_order(request: PushOrderRequest):
+        try:
+            result = await intels.push_action_order(
+                demand=request.demand,
+                prompt_id=request.prompt_id,
+                action_type=request.action_type
+            )
+            return PromptResponse(msg = "success",content=result)
+        except Exception as e:
+            raise HTTPException(
+                status_code=500, detail=f"{e}"
+            )
+
+    # 人为干预
+
+    @router.get("/registered_prompt",
+                description="获取以注册的提示词",
+                response_model=PromptResponse)
+    async def get_prompt():
+        try:
+            result = ["memorycard-format",
+            "memorycard-polish",
+            "memorycard-merge",
+            "memorycard-score",
+            "memorycard-generate-content",
+            "user-overview",
+            "user-relationship-extraction",
+            "avatar-brief",
+            "avatar-personality-extraction",
+            "avatar-desensitization",
+            "biograph-free-writer",
+            "biograph-paid-title",
+            "biograph-outline",
+            "biograph-brief",
+            "biograph-extract-person-name",
+            "biograph-extract-place",
+            "biograph-extract-material",
+            "biograph-writer"]
+
+            return PromptResponse(msg = "success",content=result)
+        except Exception as e:
+            raise HTTPException(
+                status_code=500, detail=f"{e}"
+            )
+    @router.post("/get_prompt",
+                description="获得现行提示词",
+                response_model=PromptResponse)
+    async def get_prompt(request: GetPromptRequest):
+        try:
+            async with create_async_session(intels.engine) as session:
+                result = await intels.get_prompt_safe(
+                    prompt_id=request.prompt_id,
+                    session=session
+                )
+            return PromptResponse(msg = "success",content=result)
+    
+        except Exception as e:
+            raise HTTPException(
+                status_code=500, detail=f"{e}"
+            )
+        
+    @router.post("/update_prompt",
+                description="更新现行提示词",
+                response_model=PromptResponse)
+    async def update_prompt(request: UpdatePromptRequest):
+        try:
+            async with create_async_session(intels.engine) as session:
+                await intels.save_prompt(
+                                prompt_id = request.prompt_id,
+                                new_prompt = request.prompt,
+                                use_case = "",
+                                action_type = "inference",
+                                demand = "上传",
+                                score = 70,
+                                session = session)
+                return PromptResponse(msg = "success",content=None)
+        except Exception as e:
+            raise HTTPException(
+                status_code=500, detail=f"{e}"
+            )
+
+    @router.post("/rollback_prompt",
+                description="回滚现行提示词",
+                response_model=PromptResponse)
+    async def roll_back(request: RollBackPromptRequest):
+        try:
+            async with create_async_session(intels.engine) as session:
+                result = await intels.get_prompt_safe(
+                    prompt_id=request.prompt_id,
+                    version = request.version,
+                    session=session
+                )
+                assert result.version == request.version
+                await intels.save_prompt(
+                                prompt_id = request.prompt_id,
+                                new_prompt = result.prompt,
+                                use_case = result.use_case,
+                                action_type = "inference",
+                                demand = "",
+                                score = 61,
+                                session = session)
+            return PromptResponse(msg = "success",content=None)
+        except Exception as e:
+            raise HTTPException(
+                status_code=500, detail=f"{e}"
+            )
+        
+
+    #系统级别服务
+    
+    @router.post("/sync_database")
+    async def sync_database(request: SyncDataBaseRequest):
+        try:
+            slave_database_url = request.slave_database_url or slave_database_url
+            result = await intels.sync_production_database(slave_database_url)
+            return PromptResponse(msg = "success",content=result)
+        except Exception as e:
+            raise HTTPException(
+                status_code=500, detail=f"{e}"
+            )
+    return router
+

{pro_craft-0.1.37 → pro_craft-0.1.39}/src/pro_craft.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pro-craft
-Version: 0.1.37
+Version: 0.1.39
 Summary: Add your description here
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -9,6 +9,7 @@ Requires-Dist: anyio>=4.11.0
 Requires-Dist: db-help>=0.2.2
 Requires-Dist: fastapi>=0.119.0
 Requires-Dist: llmada>=1.1.11
+Requires-Dist: mcp[cli]>=1.19.0
 Requires-Dist: plotly>=6.3.1
 Requires-Dist: pyyaml>=6.0.3
 Requires-Dist: toml>=0.10.2

{pro_craft-0.1.37 → pro_craft-0.1.39}/src/pro_craft.egg-info/SOURCES.txt

@@ -17,8 +17,10 @@ src/pro_craft/prompt_craft/async_.py
 src/pro_craft/prompt_craft/new.py
 src/pro_craft/prompt_craft/sync.py
 src/pro_craft/server/mcp/__init__.py
+src/pro_craft/server/mcp/models.py
 src/pro_craft/server/mcp/prompt.py
 src/pro_craft/server/router/__init__.py
+src/pro_craft/server/router/models.py
 src/pro_craft/server/router/prompt.py
 tests/test22.py
 tests/test_11.py

{pro_craft-0.1.37 → pro_craft-0.1.39}/src/pro_craft.egg-info/requires.txt

@@ -3,6 +3,7 @@ anyio>=4.11.0
 db-help>=0.2.2
 fastapi>=0.119.0
 llmada>=1.1.11
+mcp[cli]>=1.19.0
 plotly>=6.3.1
 pyyaml>=6.0.3
 toml>=0.10.2

pro_craft-0.1.37/src/pro_craft/server/mcp/prompt.py

@@ -1,45 +0,0 @@
-from mcp.server.fastmcp import FastMCP
-
-from pro_craft import Intel
-
-def create_mcp(database_url: str,
-                  slave_database_url: str,
-                  model_name: str):
-    # region MCP Weather
-    mcp = FastMCP("PromptManager")
-
-    intels = Intel(
-        database_url=database_url,
-        model_name=model_name
-        )
-
-    @mcp.tool()
-    def push_order(demand: str, prompt_id: str, action_type: str = "train") -> str:
-        result = intels.push_action_order(
-            demand=demand,
-            prompt_id=prompt_id,
-            action_type=action_type
-        )
-        return {"message": "success", "result": result}
-
-    @mcp.tool()
-    def get_latest_prompt(prompt_id: str) -> str:
-        with create_session(intels.engine) as session:
-            result = intels.get_prompts_from_sql(
-                prompt_id=prompt_id,
-                session=session
-            )
-        return {"message": "success", "result": result}
-
-
-    @mcp.tool()
-    def sync_database() -> str:
-        result = intels.sync_prompt_data_to_database(slave_database_url)
-        return {"message": "success","result":result}
-    
-    return mcp
-
-
-if __name__ == "__main__":
-    mcp = create_mcp()
-    mcp.run(transport="streamable-http")

pro_craft-0.1.37/src/pro_craft/server/router/prompt.py

@@ -1,80 +0,0 @@
-
-
-from fastapi import APIRouter
-from pro_craft import Intel,AsyncIntel
-from pro_craft.utils import create_async_session
-
-def create_router(database_url: str,
-                  slave_database_url: str,
-                  model_name: str,
-                  logger = None):
-    """
-    # TODO 整理改为异步
-    创建一个包含 ProCraft 路由的 FastAPI APIRouter 实例。
-
-    Args:
-        database_url (str): 数据库连接字符串。
-        model_name (str): 用于 Intel 实例的模型名称。
-        api_key_secret (str, optional): 用于验证 API Key 的秘密字符串。
-                                        如果提供，它将覆盖环境变量 PRO_CRAFT_API_KEY。
-                                        如果都不提供，会使用硬编码的 'your_default_secret_key'。
-    Returns:
-        APIRouter: 配置好的 FastAPI APIRouter 实例。
-    """
-
-    intels = AsyncIntel(
-        database_url=database_url,
-        model_name=model_name,
-        logger=logger
-        )
-
-    router = APIRouter(
-        tags=["prompt"] # 这里使用 Depends 确保每次请求都验证
-    )
-
-    @router.get("/push_order",
-                description="可选 train,inference,summary,finetune,patch",)
-    async def push_order(demand: str, prompt_id: str, action_type: str = "train"):
-        result = await intels.push_action_order(
-            demand=demand,
-            prompt_id=prompt_id,
-            action_type=action_type
-        )
-        return {"message": "success", "result": result}
-
-    @router.get("/get_latest_prompt")
-    async def get_latest_prompt(prompt_id: str):
-        async with create_async_session(intels.engine) as session:
-            result = await intels.get_prompts_from_sql(
-                prompt_id=prompt_id,
-                session=session
-            )
-        return {"message": "success", "result": result}
-
-    @router.get("/sync_database")
-    async def sync_database():
-        result = await 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):
-        async with create_async_session(intels.engine) as session:
-            result = await intels.get_prompts_from_sql(
-                prompt_id=prompt_id,
-                version = version,
-                session=session
-            )
-            assert result.version == version
-            await intels.save_prompt_increment_version(
-                            prompt_id = prompt_id,
-                            new_prompt = result.prompt,
-                            use_case = result.use_case,
-                            action_type = "inference",
-                            demand = "",
-                            score = 61,
-                            session = session)
-        return {"message": "success"}
-    
-    return router
-