pro-craft 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

pro_craft/server/mcp/content.py

@@ -1,113 +0,0 @@
-
-from mcp.server.fastmcp import Context, FastMCP, Icon
-from prompt_writing_assistant.file_manager import ContentManager, TextType
-from mcp.server.fastmcp import FastMCP
-from mcp.server.fastmcp.prompts import base
-from pydantic import BaseModel, Field
-from prompt_writing_assistant.prompt_helper import IntellectType,Intel
-
-
-content_manager = ContentManager()
-intels = Intel()
-
-mcp = FastMCP("Content")
-
-@mcp.prompt(title="Code Review")
-def review_code(code: str) -> str:
-    return f"Please review this code:\n\n{code}"
-
-
-@mcp.prompt(title="Debug Assistant")
-def debug_error(error: str) -> list[base.Message]:
-    return [
-        base.UserMessage("I'm seeing this error:"),
-        base.UserMessage(error),
-        base.AssistantMessage("I'll help debug that. What have you tried so far?"),
-    ]
-
-
-@mcp.tool()
-def save_content(text:str)->str:
-    """
-    text : 需要存储的内容
-    return : 返回是否存储成功的信息
-    """
-    result =content_manager.save_content_auto(
-                text = text)
-    return result
-
-@mcp.tool()
-def similarit_content(text: str,limit: int):
-    """
-    text : 待查询的文字, 进行匹配
-    limit : 查询的数量
-    """
-    
-    result = content_manager.similarity(
-        content = text,
-        limit = limit
-    )
-    return result
-
-@mcp.tool()
-def get_prompts(prompt_id: str,version: str = "1.0"):
-    """
-    prompt_id : 待查询的文字, 进行匹配
-    limit : 查询的数量
-    """
-    result = intels.get_prompts_from_sql(
-        table_name = "prompts_data",
-        prompt_id = prompt_id,
-        version=version,
-        )
-    return result
-
-@mcp.tool()
-def list_cities() -> list[str]:
-    """Get a list of cities"""
-    return ["London", "Paris", "Tokyo"]
-
-
-
-
-
-
-class WeatherData(BaseModel):
-    """Weather information structure."""
-
-    temperature: float = Field(description="Temperature in Celsius")
-    humidity: float = Field(description="Humidity percentage")
-    condition: str
-    wind_speed: float
-
-
-@mcp.tool() # icons=[icon]
-def get_weather(city: str) -> WeatherData:
-    """Get weather for a city - returns structured data."""
-    # Simulated weather data
-    return WeatherData(
-        temperature=22.5,
-        humidity=45.0,
-        condition="sunny",
-        wind_speed=5.2,
-    )
-
-
-from PIL import Image as PILImage
-from mcp.server.fastmcp import FastMCP, Image
-
-@mcp.tool()
-def create_thumbnail(image_path: str) -> Image:
-    """Create a thumbnail from an image"""
-    img = PILImage.open(image_path)
-    img.thumbnail((100, 100))
-    return Image(data=img.tobytes(), format="png")
-
-
-
-if __name__ == "__main__":
-    mcp.run(transport="streamable-http")
-    # mcp.run(transport="sse")
-    # search_mcp.run(transport="sse", mount_path="/search")
-
-

pro_craft/server/mcp/math.py

@@ -1,33 +0,0 @@
-from mcp.server.fastmcp import FastMCP
-
-
-# region MCP Math
-mcp = FastMCP("Math")
-
-@mcp.tool(description="A simple add tool")
-def add(a: int, b: int) -> int:
-    """Adds two integers.
-       Args:
-         a: The first integer.
-         b: The second integer.
-    """
-    return a + b
-
-
-@mcp.tool()
-def multiply(a: int, b: int) -> int:
-    """Multiply two integers.
-       Args:
-         a: The first integer.
-         b: The second integer.
-    """
-    return a * b
-# endregion
-
-
-if __name__ == "__main__":
-    mcp.run(transport="streamable-http")
-    # mcp.run(transport="sse")
-    # search_mcp.run(transport="sse", mount_path="/search")
-
-

pro_craft/server/mcp/resource.py

@@ -1,35 +0,0 @@
-
-from mcp.server.fastmcp import Context, FastMCP
-
-
-# region MCP Math
-mcp = FastMCP("resource")
-
-
-@mcp.resource("config://settings")
-def get_settings() -> str:
-    """Get application settings."""
-    return """{
-  "theme": "dark",
-  "language": "en",
-  "debug": false
-}"""
-
-@mcp.resource("file://documents/{name}")
-def read_document(name: str) -> str:
-    """Read a document by name."""
-    # This would normally read from disk
-    path = "/Users/zhaoxuefeng/GitHub/prompt_writing_assistant/tests/temp_file/"
-    path += name
-    with open(path,'r') as f:
-        return f.read()
-
-@mcp.resource("file://notes/{name}")
-def read_notes(name:str)-> str:
-    """ 阅读备忘录"""
-
-    return "你要做王子"
-
-if __name__ == "__main__":
-    mcp.run(transport="streamable-http")
-

pro_craft/server/models.py

@@ -1,6 +0,0 @@
-
-# server
-from typing import Dict, Any, Optional, List
-from pydantic import BaseModel, Field, model_validator
-
-

pro_craft/server/router/recommended.py

@@ -1,283 +0,0 @@
-# server
-# 推荐算法
-
-from ..models import UpdateItem, DeleteResponse, DeleteRequest, QueryItem
-from fastapi import APIRouter, Depends, HTTPException, status
-from typing import Dict, Any
-from diglife.embedding_pool import EmbeddingPool
-from diglife.log import Log
-import os
-import httpx
-
-from dotenv import load_dotenv, find_dotenv
-
-dotenv_path = find_dotenv()
-load_dotenv(dotenv_path, override=True)
-
-
-
-router = APIRouter(tags=["recommended"])
-
-logger = Log.logger
-
-recommended_biographies_cache_max_leng = os.getenv("recommended_biographies_cache_max_leng",2) #config.get("recommended_biographies_cache_max_leng", 2)
-recommended_biographies_cache_max_leng = int(recommended_biographies_cache_max_leng)
-recommended_cache_max_leng = os.getenv("recommended_cache_max_leng",2) #config.get("recommended_cache_max_leng", 2)
-recommended_cache_max_leng = int(recommended_cache_max_leng)
-user_server_base_url = "http://182.92.107.224:7000"
-
-ep = EmbeddingPool()
-recommended_biographies_cache: Dict[str, Dict[str, Any]] = {}
-recommended_figure_cache: Dict[str, Dict[str, Any]] = {}
-
-@router.post(
-    "/update",  # 推荐使用POST请求进行数据更新
-    summary="更新或添加文本嵌入",
-    description="将给定的文本内容与一个ID关联并更新到Embedding池中。",
-    response_description="表示操作是否成功。",
-)
-def recommended_update(item: UpdateItem):
-    """记忆卡片是0  传记是1
-    记忆卡片是0
-    记忆卡片上传的是记忆卡片的内容 str
-    记忆卡片id
-    0
-
-    传记是1
-    上传的是传记简介  str
-    传记id
-    1
-
-    数字分身是2
-    上传数字分身简介和性格描述  str
-    数字分身id
-    2
-    """
-    # TODO 需要一个反馈状态
-    try:
-        if item.type in [0, 1, 2]:  # 上传的是卡片
-            ep.update(text=item.text, id=item.id, type=item.type)
-        else:
-            logger.error(f"Error updating EmbeddingPool for ID '{item.id}': {e}")
-            raise HTTPException(
-                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-                detail=f"Failed to update embedding for ID '{item.id}': {e}",
-            )
-
-        return {"status": "success", "message": f"ID '{item.id}' updated successfully."}
-
-    except ValueError as e:  # 假设EmbeddingPool.update可能抛出ValueError
-        logger.warning(f"Validation error during update: {e}")
-        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(e))
-    except Exception as e:
-        logger.error(f"Error updating EmbeddingPool for ID '{item.id}': {e}")
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"Failed to update embedding for ID '{item.id}': {e}",
-        )
-
-
-@router.post("/delete", response_model=DeleteResponse, description="delete")
-async def delete_server(request: DeleteRequest):
-
-    logger.info("running delete_server")
-
-    # TODO 需要一个反馈状态
-    result = ep.delete(id=request.id)  # 包裹的内核函数
-
-    ########
-    return DeleteResponse(
-        status="success",
-    )
-
-
-
-
-# async def aget_content_by_id(url = ""):
-#     # url = url.format(user_profile_id = user_profile_id)
-#     async with httpx.AsyncClient() as client:
-#         try:
-#             response = await client.get(url)
-#             response.raise_for_status()  # 如果状态码是 4xx 或 5xx，会抛出 HTTPStatusError 异常
-            
-#             print(f"Status Code: {response.status_code}")
-#             print(f"Response Body: {response.json()}") # 假设返回的是 JSON
-#             return response.json()
-#         except httpx.HTTPStatusError as e:
-#             print(f"HTTP error occurred: {e.response.status_code} - {e.response.text}")
-#         except httpx.RequestError as e:
-#             print(f"An error occurred while requesting {e.request.url!r}: {e}")
-#         except Exception as e:
-#             print(f"An unexpected error occurred: {e}")
-#     return None
-
-async def aget_(url = ""):
-    async with httpx.AsyncClient() as client:
-        try:
-            response = await client.get(url)
-            response.raise_for_status()  # 如果状态码是 4xx 或 5xx，会抛出 HTTPStatusError 异常
-            
-            print(f"Status Code: {response.status_code}")
-            print(f"Response Body: {response.json()}") # 假设返回的是 JSON
-            return response.json()
-        except httpx.HTTPStatusError as e:
-            print(f"HTTP error occurred: {e.response.status_code} - {e.response.text}")
-        except httpx.RequestError as e:
-            print(f"An error occurred while requesting {e.request.url!r}: {e}")
-        except Exception as e:
-            print(f"An unexpected error occurred: {e}")
-    return None
-
-@router.post(
-    "/search_biographies_and_cards",
-    summary="搜索传记和记忆卡片",
-    description="搜索传记和记忆卡片",
-    response_description="搜索结果列表。",
-)
-async def recommended_biographies_and_cards(query_item: QueryItem):
-    """
-        # result = [
-        #     {
-        #         "id": "1916693308020916225",  # 传记ID
-        #         "type": 1,
-        #         "order": 0,
-        #     },
-        #     {
-        #         "id": "1962459564012359682",  # 卡片ID
-        #         "type": 0,
-        #         "order": 1,
-        #     },
-        #     {
-        #         "id": "1916389315373727745",  # 传记ID
-        #         "type": 1,
-        #         "order": 2,
-        #     },
-        # ]
-
-        {
-        "text":"这是一个传记001",
-        "id":"1916693308020916225",
-        "type":1
-    }
-    {
-        "text":"这是一个传记002",
-        "id":"1916389315373727745",
-        "type":1
-    }
-    {
-        "text":"这是一个卡片001",
-        "id":"1962459564012359682",
-        "type":0
-    }
-    """
-    try:
-        # TODO 需要一个通过id 获取对应内容的接口
-        # TODO 调用id 获得对应的用户简介 query_item.user_id
-
-
-        user_profile_id_to_fetch = query_item.user_id
-        # memory_info = await aget_content_by_id(user_profile_id_to_fetch,url = user_server_base_url + "/api/inner/getMemoryCards?userProfileId={user_profile_id}")
-        memory_info = await aget_(url = user_server_base_url + f"/api/inner/getMemoryCards?userProfileId={user_profile_id_to_fetch}")
-        # memory_info = await get_memorycards_by_id(user_profile_id_to_fetch)
-        user_brief = '\n'.join([i.get('content') for i in memory_info['data']["memoryCards"][:4]])
-
-
-        result = ep.search_bac(query=user_brief)
-
-        if recommended_biographies_cache.get(query_item.user_id):
-            clear_result = [
-                i
-                for i in result
-                if i.get("id")
-                not in recommended_biographies_cache.get(query_item.user_id)
-            ]
-        else:
-            recommended_biographies_cache[query_item.user_id] = []
-            clear_result = result
-
-        recommended_biographies_cache[query_item.user_id] += [
-            i.get("id") for i in result
-        ]
-        recommended_biographies_cache[query_item.user_id] = list(
-            set(recommended_biographies_cache[query_item.user_id])
-        )
-        if (
-            len(recommended_biographies_cache[query_item.user_id])
-            > recommended_biographies_cache_max_leng
-        ):
-            recommended_biographies_cache[query_item.user_id] = []
-
-        return {
-            "status": "success",
-            "result": clear_result,
-            "query": query_item.user_id,
-        }
-
-    except Exception as e:
-        logger.error(
-            f"Error searching EmbeddingPool for query '{query_item.user_id}': {e}"
-        )
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"Failed to perform search: {e}",
-        )
-
-
-
-@router.post(
-    "/search_figure_person",
-    description="搜索数字分身的",
-)
-async def recommended_figure_person(query_item: QueryItem):
-    """
-
-    """
-    try:
-
-        user_profile_id_to_fetch = query_item.user_id
-        # avatar_info = await aget_avatar_desc_by_id(user_profile_id_to_fetch)
-        # avatar_info = await aget_content_by_id(user_profile_id_to_fetch,url = user_server_base_url + "/api/inner/getAvatarDesc?userProfileId={user_profile_id}")
-        avatar_info = await aget_(url = user_server_base_url + f"/api/inner/getAvatarDesc?userProfileId={user_profile_id_to_fetch}")
-        print(avatar_info,'avatar_info')
-        if avatar_info["code"] == 200:
-            user_brief = avatar_info["data"].get("avatarDesc")
-        else:
-            user_brief = "这是一个简单的人"
-
-        result = ep.search_figure_person(query=user_brief)  # 100+
-
-        if recommended_figure_cache.get(query_item.user_id):
-            # 不需要创建
-            clear_result = [
-                i
-                for i in result
-                if i.get("id") not in recommended_figure_cache.get(query_item.user_id)
-            ]
-        else:
-            recommended_figure_cache[query_item.user_id] = []
-            clear_result = result
-
-        recommended_figure_cache[query_item.user_id] += [i.get("id") for i in result]
-        recommended_figure_cache[query_item.user_id] = list(
-            set(recommended_figure_cache[query_item.user_id])
-        )
-        if (
-            len(recommended_figure_cache[query_item.user_id])
-            > recommended_cache_max_leng
-        ):
-            recommended_figure_cache[query_item.user_id] = []
-        return {
-            "status": "success",
-            "result": clear_result,
-            "query": query_item.user_id,
-        }
-
-    except Exception as e:
-        logger.error(
-            f"Error searching EmbeddingPool for query '{query_item.user_id}': {e}"
-        )
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"Failed to perform search: {e}",
-        )
-

pro_craft/server.py

@@ -1,100 +0,0 @@
-
-from fastapi import FastAPI, HTTPException, Header
-from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel, Field, model_validator
-import argparse
-import uvicorn
-from .log import Log
-
-default=8008
-
-app = FastAPI(
-    title="LLM Service",
-    description="Provides an OpenAI-compatible API for custom large language models.",
-    version="1.0.1",
-)
-
-# --- Configure CORS ---
-origins = [
-    "*", # Allows all origins (convenient for development, insecure for production)
-    # Add the specific origin of your "别的调度" tool/frontend if known
-    # e.g., "http://localhost:5173" for a typical Vite frontend dev server
-    # e.g., "http://127.0.0.1:5173"
-]
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=origins,  # Specifies the allowed origins
-    allow_credentials=True, # Allows cookies/authorization headers
-    allow_methods=["*"],    # Allows all methods (GET, POST, OPTIONS, etc.)
-    allow_headers=["*"],    # Allows all headers (Content-Type, Authorization, etc.)
-)
-# --- End CORS Configuration ---
-
-
-@app.get("/")
-async def root():
-    """ x """
-    return {"message": "LLM Service is running."}
-
-
-if __name__ == "__main__":
-    
-    parser = argparse.ArgumentParser(
-        description="Start a simple HTTP server similar to http.server."
-    )
-    parser.add_argument(
-        'port',
-        metavar='PORT',
-        type=int,
-        nargs='?', # 端口是可选的
-        default=default,
-        help=f'Specify alternate port [default: {default}]'
-    )
-    # 创建一个互斥组用于环境选择
-    group = parser.add_mutually_exclusive_group()
-
-    # 添加 --dev 选项
-    group.add_argument(
-        '--dev',
-        action='store_true', # 当存在 --dev 时，该值为 True
-        help='Run in development mode (default).'
-    )
-
-    # 添加 --prod 选项
-    group.add_argument(
-        '--prod',
-        action='store_true', # 当存在 --prod 时，该值为 True
-        help='Run in production mode.'
-    )
-    args = parser.parse_args()
-
-    if args.prod:
-        env = "prod"
-    else:
-        # 如果 --prod 不存在，默认就是 dev
-        env = "dev"
-
-    port = args.port
-    if env == "dev":
-        port += 100
-        Log.reset_level('debug',env = env)
-        reload = True
-        app_import_string = f"{__package__}.server:app" # <--- 关键修改：传递导入字符串
-    elif env == "prod":
-        Log.reset_level('info',env = env)# ['debug', 'info', 'warning', 'error', 'critical']
-        reload = False
-        app_import_string = app
-    else:
-        reload = False
-        app_import_string = app
-    
-
-    # 使用 uvicorn.run() 来启动服务器
-    # 参数对应于命令行选项
-    uvicorn.run(
-        app_import_string,
-        host="0.0.0.0",
-        port=port,
-        reload=reload  # 启用热重载
-    )

pro_craft-0.1.12.dist-info/RECORD

@@ -1,24 +0,0 @@
-pro_craft/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pro_craft/database.py,sha256=5dV-h9dVaS6euHLCtf0gYfq2pchl2QFdb2PEM4gTEU4,8740
-pro_craft/designer.py,sha256=3gyCqrjcw61sHzDjUPKhL1LOAE8xWLLbNT8NlK2mFLc,4739
-pro_craft/evals.py,sha256=1T86jur4k3cLk43j1GyAW4JS0nPNfl6P0ZOQmu-SgpA,1928
-pro_craft/file_manager.py,sha256=2j7lCt9L4mtvAy8_76ibTthXLwKKmVatWIB3DSvQM7U,3805
-pro_craft/log.py,sha256=MZf9jCZsiRoAq8v4FxVnJqeSXxgzAiiKf7mxz6bFtwM,4263
-pro_craft/prompt_helper.py,sha256=GClPsomdGdo6FXlp8VdiQy8GaBtj3oLeq2oFw0RiM3k,26592
-pro_craft/prompt_helper_async.py,sha256=-EtwjDUL2Xoi4qjZpzgnUyAVBkt-WCocGgVl5PgA4Js,27771
-pro_craft/prompt_helper_new.py,sha256=DV871XElJmHDO2xis1mC1f3QFvq3ll81VGGD9aj0Eiw,23542
-pro_craft/server.py,sha256=fPAosQIU0d7gxICiALl8u6QwbLI4cawVFyoRYebRES0,2827
-pro_craft/utils.py,sha256=cpvwk68mD9hYY8WCq2JXzfrrXqpshiscz_OSav4tC7U,5687
-pro_craft/code_helper/coder.py,sha256=NXglF1KiPtGe4HZN0MZvFJ8p9Iyd5kzIt72DQGgRwXA,24715
-pro_craft/server/__main__.py,sha256=LDTERPMe7RKj3eifVRo9aO9fNXdd16W5Hzr1APd04T0,4227
-pro_craft/server/models.py,sha256=CiUK8e73Bl7fo7ZbnwNTLYLeD4pb1fHMzWR13d3Y6vs,112
-pro_craft/server/mcp/__init__.py,sha256=4dbl-lFcm0r2tkOP04OxqiZG2jR-rqF181qi2AfU6UA,123
-pro_craft/server/mcp/content.py,sha256=DombC1DlasvEl_lzy4u6soVR7rxgmyWFlaEvTSdCiOU,2745
-pro_craft/server/mcp/math.py,sha256=OOzGXx64nK4bOVlu33PtVddcCQ9ilqA3Em9yxjSX9cg,641
-pro_craft/server/mcp/resource.py,sha256=z94jP3qZofO-1lZCM3TuOfLajw41HARs1ojXab1ymas,776
-pro_craft/server/mcp/weather.py,sha256=RAGuf4sgjlTQSfRRZ1Fo18JnuMQRS_Db9p6AqBQrl8E,455
-pro_craft/server/router/recommended.py,sha256=IAZFdmb8HSl2_TOJeuv5uOKIX47XyX4p4sEwxG-0vt0,9968
-pro_craft-0.1.12.dist-info/METADATA,sha256=VKEqBoEWYaOP0Ue2BSZ3iKZ1W_MYqsaRS7f3lQ-bscs,1800
-pro_craft-0.1.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pro_craft-0.1.12.dist-info/top_level.txt,sha256=yqYDHArnYMWpeCxkmGRwlL6sJtxiOUnYylLDx9EOgFg,10
-pro_craft-0.1.12.dist-info/RECORD,,