pro-craft 0.1.1__py3-none-any.whl

pro_craft/log.py

@@ -0,0 +1,118 @@
+import logging
+import os
+from logging.handlers import RotatingFileHandler, TimedRotatingFileHandler
+
+class Logger:
+    """log
+
+    Returns:
+        _type_: single
+    """
+    _instance = None
+
+    def __new__(cls, *args, **kwargs):
+        if cls._instance is None:
+            cls._instance = super().__new__(cls)
+        return cls._instance
+
+    def __init__(self,level = 'debug',log_file_name = "app.log"):
+        """
+        Level (级别): 定义日志的严重程度。从低到高依次是：
+        DEBUG: 详细的调试信息，通常只在开发阶段使用。
+        INFO: 确认程序按预期运行。
+        WARNING: 发生了意外，或将来可能出现问题，但程序仍在正常运行。
+        ERROR: 发生了严重错误，程序可能无法执行某些功能。
+        CRITICAL: 发生了严重错误，程序可能无法继续运行。
+                # --- 1. 定义日志级别 ---
+        # 建议在开发环境使用 DEBUG，生产环境使用 INFO 或 WARNING
+
+        """
+        if not hasattr(self, 'initialized'):
+            self.initialized = True
+
+            if level == 'debug':
+                self.LOG_LEVEL = logging.DEBUG  # 开发阶段
+            elif level == 'info':
+                self.LOG_LEVEL = logging.INFO  # 生产阶段
+            elif level == 'warning':
+                self.LOG_LEVEL = logging.WARNING
+            elif level == 'error':
+                self.LOG_LEVEL = logging.ERROR
+            elif level == 'critical':
+                self.LOG_LEVEL = logging.CRITICAL
+            else:
+                self.LOG_LEVEL = logging.INFO  # 默认级别
+
+            # --- 2. 定义日志文件路径和名称 ---
+            self.LOG_DIR = "logs"
+            self.LOG_FILE_NAME = log_file_name
+            self.LOG_FILE_PATH = os.path.join(self.LOG_DIR, self.LOG_FILE_NAME)
+
+            # 确保日志目录存在
+            os.makedirs(self.LOG_DIR, exist_ok=True)
+            self.logger = None
+            self.setup_logging()
+            self.env = 'dev'
+
+    def reset_level(self,level = 'debug',env = 'dev'):
+        if level == 'debug':
+            self.LOG_LEVEL = logging.DEBUG  # 开发阶段
+        elif level == 'info':
+            self.LOG_LEVEL = logging.INFO  # 生产阶段
+        elif level == 'warning':
+            self.LOG_LEVEL = logging.WARNING
+        elif level == 'error':
+            self.LOG_LEVEL = logging.ERROR
+        elif level == 'critical':
+            self.LOG_LEVEL = logging.CRITICAL
+        else:
+            self.LOG_LEVEL = logging.INFO  # 默认级别
+
+        self.setup_logging()
+        self.env = env
+
+    def setup_logging(self):
+        """# --- 3. 配置 Logger ---
+        """
+        # 获取根 Logger (也可以创建自定义的 Logger: logging.getLogger('my_app'))
+        logger = logging.getLogger()
+        logger.setLevel(self.LOG_LEVEL)
+
+        # 避免重复添加 Handler (如果多次调用 setup_logging)
+        if not logger.handlers:
+            # --- 4. 配置 Formatter (格式化器) ---
+            # 常见格式：时间 - 日志级别 - Logger名称 - 模块名 - 行号 - 消息
+            formatter = logging.Formatter(
+                '%(asctime)s - %(levelname)s - %(name)s - %(module)s:%(lineno)d - %(message)s'
+            )
+
+            # --- 5. 配置 Handler (处理器) ---
+
+            # 5.1 控制台处理器 (StreamHandler)
+            console_handler = logging.StreamHandler()
+            console_handler.setLevel(logging.INFO) # 控制台只显示 INFO 及以上级别的日志
+            console_handler.setFormatter(formatter)
+            logger.addHandler(console_handler)
+
+            # 5.2 文件处理器 (RotatingFileHandler 或 TimedRotatingFileHandler)
+
+            file_handler = RotatingFileHandler(
+                self.LOG_FILE_PATH,
+                maxBytes=10 * 1024 * 1024, # 10 MB
+                backupCount=5,
+                encoding='utf-8'
+            )
+            file_handler.setLevel(self.LOG_LEVEL) # 文件中显示所有指定级别的日志
+            file_handler.setFormatter(formatter)
+            logger.addHandler(file_handler)
+
+        self.logger = logger
+
+
+Log = Logger(log_file_name = "app.log")
+del Logger
+
+"""
+from .log import Log
+logger = Log.logger
+"""

pro_craft/server/__main__.py

@@ -0,0 +1,160 @@
+
+from fastapi import FastAPI, HTTPException, Header
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI, Depends
+from fastapi_users import FastAPIUsers
+from contextlib import asynccontextmanager
+from .utils import create_db_and_tables
+from .utils import get_user_manager, auth_backend
+from .models.models import User,UserCreate, UserRead,UserUpdate
+from .routers import admin_router,user_router
+from pydantic import BaseModel, Field, model_validator
+import argparse
+import uvicorn
+import uuid
+from umanager.log import Log
+
+# 应用程序生命周期事件
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # 应用启动时创建数据库表
+    await create_db_and_tables()
+    yield
+
+
+default=8008
+
+app = FastAPI(
+    lifespan=lifespan,
+    title="LLM Service",
+    description="Provides an OpenAI-compatible API for custom large language models.",
+    version="1.0.1",
+)
+
+fastapi_users = FastAPIUsers[User, uuid.UUID](
+    get_user_manager,
+    [auth_backend],
+)
+
+
+# --- 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.include_router(
+    fastapi_users.get_auth_router(auth_backend),
+    prefix="/auth/jwt",
+    tags=["auth"],
+)
+app.include_router(
+    fastapi_users.get_register_router(UserRead, UserCreate),
+    prefix="/auth",
+    tags=["auth"],
+)
+app.include_router(
+    fastapi_users.get_reset_password_router(),
+    prefix="/auth",
+    tags=["auth"],
+)
+app.include_router(
+    fastapi_users.get_verify_router(UserRead),
+    prefix="/auth",
+    tags=["auth"],
+)
+app.include_router(
+    fastapi_users.get_users_router(UserRead, UserUpdate),
+    prefix="/users",
+    tags=["users"],
+)
+
+
+# -------------------- 集成待办事项路由 --------------------
+# app.include_router(admin_router, prefix="/")
+# app.include_router(user_router, prefix="/")
+
+app.include_router(admin_router,prefix="/admin")
+app.include_router(user_router,prefix="/users")
+
+
+@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__}.__main__: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/server/mcp/__init__.py

@@ -0,0 +1,5 @@
+# uv run mcp dev src/obsidian_sdk/mcp.py
+# TODO 开发mcp 的stdio 格式, 使得MCP 更加好用
+# Create an MCP server
+
+

pro_craft/server/mcp/content.py

@@ -0,0 +1,113 @@
+
+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

@@ -0,0 +1,33 @@
+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

@@ -0,0 +1,35 @@
+
+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/mcp/weather.py

@@ -0,0 +1,22 @@
+from datetime import datetime
+from mcp.server.fastmcp import FastMCP
+
+# region MCP Weather
+mcp = FastMCP("Weather")
+
+@mcp.tool()
+def get_weather(location: str) -> str:
+    return "Cloudy"
+
+@mcp.tool()
+def get_time() -> str:
+    return datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+# endregion
+
+
+if __name__ == "__main__":
+    mcp.run(transport="streamable-http")
+    # mcp.run(transport="sse")
+    # search_mcp.run(transport="sse", mount_path="/search")
+
+

pro_craft/server/models/models.py

@@ -0,0 +1,48 @@
+
+import uuid
+from typing import Optional
+
+from sqlalchemy import String, Boolean, Column
+from sqlalchemy.dialects.mysql import CHAR # Use CHAR for UUID to store as fixed length string
+from fastapi_users.db import SQLAlchemyBaseUserTableUUID
+from pydantic import BaseModel, Field
+
+from umanager.server.utils import Base # 引入Base
+
+# SQLAlchemy 用户模型
+# 使用 SQLAlchemyBaseUserTableUUID 来自动处理 UUID 主键
+class User(SQLAlchemyBaseUserTableUUID, Base):
+    __tablename__ = "users" # 表名
+    
+    # 可以在这里添加额外的用户属性
+    first_name: str = Column(String(255), nullable=True)
+    last_name: str = Column(String(255), nullable=True)
+    is_admin: bool = Column(Boolean, default=False)
+
+# Pydantic 模型 (用于请求和响应)
+class UserRead(BaseModel):
+    id: uuid.UUID
+    email: str
+    is_active: bool = True
+    is_superuser: bool = False
+    is_verified: bool = False
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
+    is_admin: bool = False
+
+    class Config:
+        from_attributes = True # updated from orm_mode = True
+
+class UserCreate(BaseModel):
+    email: str
+    password: str
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
+
+class UserUpdate(UserCreate):
+    is_active: Optional[bool] = None
+    is_superuser: Optional[bool] = None
+    is_verified: Optional[bool] = None
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
+    is_admin: Optional[bool] = None

pro_craft/server/routers/__init__.py

@@ -0,0 +1,2 @@
+from .admin import router as admin_router
+from .user import router as user_router

pro_craft/server/routers/admin.py

@@ -0,0 +1,42 @@
+
+# app/routers/admin.py
+from fastapi import APIRouter, Depends, HTTPException, status
+from fastapi import FastAPI, HTTPException, Header
+
+
+# 模拟获取当前用户的依赖项
+async def get_current_user(x_token: str = Header(...)):
+    if x_token != "valid-secret-token":
+        raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid X-Token header")
+    return {"username": "admin", "roles": ["user", "admin"]}
+
+# 模拟一个管理员权限检查的依赖项
+async def verify_admin_role(user: dict = Depends(get_current_user)):
+    if "admin" not in user.get("roles", []):
+        raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="Not enough permissions")
+    return user
+
+
+router = APIRouter(
+    tags=["Admin"],
+    dependencies=[Depends(get_current_user), Depends(verify_admin_role)] # 统一的依赖项列表
+)
+
+
+
+@router.get("/dashboard/")
+async def get_admin_dashboard_data(user):
+    # 这个函数会自动接收到 get_current_user 和 verify_admin_role 的结果（如果需要）
+    # 但我们在这里不需要显式接收它们，因为它们只是做权限检查
+    print(user,'user')
+    return {"message": "Welcome to the admin dashboard!"}
+
+@router.post("/settings/")
+async def update_admin_settings(settings: dict):
+    # 这个函数也会自动执行 get_current_user 和 verify_admin_role
+    return {"message": "Admin settings updated", "settings": settings}
+
+# 这个API也需要认证和管理员权限
+@router.delete("/users/{user_id}")
+async def delete_user(user_id: int):
+    return {"message": f"User {user_id} deleted by admin."}

pro_craft/server/routers/user.py

@@ -0,0 +1,24 @@
+
+# app/routers/admin.py
+from fastapi import APIRouter, Depends, HTTPException, status
+from fastapi import FastAPI, Header
+
+
+# 模拟获取当前用户的依赖项
+async def get_current_user(x_token: str = Header(...)):
+    if x_token not in ["1234",
+                       "5678"]:
+        raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid X-Token header")
+    return {"username": "admin", "roles": ["user", "admin"]}
+
+
+router = APIRouter(
+    tags=["users"],
+    dependencies=[Depends(get_current_user)] # 统一的依赖项列表
+)
+
+
+@router.get("/dashboard")
+async def get_user(user_info :dict =  Depends(get_current_user)):
+    pass
+    return {"message": "Welcome to the admin dashboard!"}

pro_craft/server/utils/__init__.py

@@ -0,0 +1,3 @@
+from .database import Base, create_db_and_tables
+from .auth_backends import auth_backend
+from .user_manager import get_user_manager

pro_craft/server/utils/auth_backends.py

@@ -0,0 +1,15 @@
+# auth_backends.py
+from fastapi_users.authentication import BearerTransport, JWTStrategy, AuthenticationBackend
+
+from .user_manager import SECRET # 从 user_manager 引入 SECRET
+
+bearer_transport = BearerTransport(tokenUrl="auth/jwt/login")
+
+def get_jwt_strategy() -> JWTStrategy:
+    return JWTStrategy(secret=SECRET, lifetime_seconds=3600)  # JWT有效期1小时
+
+auth_backend = AuthenticationBackend(
+    name="jwt",
+    transport=bearer_transport,
+    get_strategy=get_jwt_strategy,
+)

pro_craft/server/utils/database.py

@@ -0,0 +1,27 @@
+# database.py
+from typing import AsyncGenerator
+
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
+from sqlalchemy.orm import sessionmaker, DeclarativeBase
+
+# 您的 MySQL 连接字符串
+# 格式：mysql+aiomysql://user:password@host:port/database
+# 注意：fastapi_users 的 sqlalchemy 适配器通常需要异步驱动，
+# 所以我们使用 `aiomysql` (虽然你安装的是 mysql-connector-python，
+# 但为了异步，我们实际会用 aiomysql，请确保安装它：pip install aiomysql)
+DATABASE_URL = "mysql+aiomysql://root:1234@localhost:3306/prompts"
+
+class Base(DeclarativeBase):
+    pass
+
+engine = create_async_engine(DATABASE_URL)
+async_session_maker = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
+
+async def get_async_session() -> AsyncGenerator[AsyncSession, None]:
+    async with async_session_maker() as session:
+        yield session
+
+# 创建所有定义的表（在应用启动时调用）
+async def create_db_and_tables():
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)

pro_craft/server/utils/user_manager.py

@@ -0,0 +1,87 @@
+# user_manager.py
+import uuid
+from typing import Optional
+
+from fastapi import Depends, Request
+from fastapi_users import BaseUserManager, UUIDIDMixin
+# from fastapi_users_sqlalchemy import SQLAlchemyUserDatabase
+from fastapi_users.db import SQLAlchemyUserDatabase
+from .database import get_async_session
+
+from fastapi import Depends, Request, HTTPException, status # <-- 确保 HTTPException 在这里
+from sqlalchemy.exc import IntegrityError # <-- 确保这一行存在
+
+from umanager.server.models.models import User, UserCreate # <--- 在这里添加 UserCreate 的导入
+
+from passlib.context import CryptContext # 确保这里引入了 CryptContext
+
+# ---------- 确保 pwd_context 在模块级别定义 ----------
+# 它应该在任何使用它的类或函数之外
+pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
+
+# 替换为您的安全密钥，最好从环境变量中获取
+SECRET = "YOUR_SUPER_SECRET_KEY_REPLACE_ME" # !!! 重要：在生产环境中，请务必使用环境变量
+
+
+class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]):
+    # ... (reset_password_token_secret, verification_token_secret 保持不变)
+
+    async def create(
+        self,
+        user_create: UserCreate,
+        safe: bool = False,
+        request: Optional[Request] = None,
+    ) -> User:
+        existing_user = await self.user_db.get_by_email(user_create.email)
+        if existing_user is not None:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="A user with this email already exists."
+            )
+
+        hashed_password = pwd_context.hash(user_create.password)
+
+        # 使用 model_dump 获取 Pydantic V2 的字典表示
+        user_data = user_create.model_dump(exclude_unset=True)
+        user_data["hashed_password"] = hashed_password
+        user_data["id"] = uuid.uuid4()
+
+        if "password" in user_data: # 确保移除明文密码
+            del user_data["password"]
+
+        # 创建 SQLAlchemy User 实例
+        # 这里的 self.user_db.user_table 就是 models.User
+        new_user = self.user_db.user_table(**user_data) 
+        
+        # 确保默认字段被设置
+        new_user.is_active = True
+        new_user.is_superuser = False
+        new_user.is_verified = False
+
+        # created_user = await self.user_db.create(new_user)
+        try:
+            self.user_db.session.add(new_user)
+            await self.user_db.session.commit()
+            await self.user_db.session.refresh(new_user) # 刷新以获取可能由数据库生成的字段（如默认值）
+        except IntegrityError:
+            await self.user_db.session.rollback()
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="A database integrity error occurred (e.g., duplicate email)."
+            )
+        # await self.on_after_register(created_user, request)
+        # return created_user
+        await self.on_after_register(new_user, request) # 这里也是 new_user
+        return new_user
+
+async def get_user_db(session = Depends(get_async_session)):
+    yield SQLAlchemyUserDatabase(session, User)
+
+
+
+async def get_user_manager(user_db: SQLAlchemyUserDatabase = Depends(get_user_db)):
+    yield UserManager(user_db)
+
+
+# uv venv .venv --python 3.11 
+# python -c "import pydantic; print(pydantic.VERSION)"

pro_craft/server.py

@@ -0,0 +1,100 @@
+
+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/utils.py

@@ -0,0 +1 @@
+from utils_tool import *

pro_craft-0.1.1.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: pro-craft
+Version: 0.1.1
+Summary: Add your description here
+Requires-Python: >=3.13
+Description-Content-Type: text/markdown
+Requires-Dist: anyio>=4.11.0
+Requires-Dist: fastapi>=0.119.0
+Requires-Dist: pytest>=8.4.2
+Requires-Dist: pytest-asyncio>=1.2.0
+Requires-Dist: pytest-tornasync>=0.6.0.post2
+Requires-Dist: toml>=0.10.2
+Requires-Dist: utils-tool==0.1.3
+Requires-Dist: uvicorn>=0.38.0

pro_craft-0.1.1.dist-info/RECORD

@@ -0,0 +1,25 @@
+pro_craft/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pro_craft/core.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pro_craft/log.py,sha256=MZf9jCZsiRoAq8v4FxVnJqeSXxgzAiiKf7mxz6bFtwM,4263
+pro_craft/server.py,sha256=fPAosQIU0d7gxICiALl8u6QwbLI4cawVFyoRYebRES0,2827
+pro_craft/utils.py,sha256=Pi6-tconaDYaYV5E-Bb-gt9jDasqcVnNBAjpdXtGY5I,25
+pro_craft/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pro_craft/server/__main__.py,sha256=e4Ygx01-YHkgk0AJTHdf7N--q1_7XwxtfS2ba5yPViM,4331
+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/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pro_craft/server/models/models.py,sha256=VlYy3VYuIAZ8p8vd5Qcnyjzy8-EnQQbbVPl6YN1mCjg,1491
+pro_craft/server/routers/__init__.py,sha256=mnJ7XmAzyZePf6CxusxbTZZF0xbqnDqYg3Z929rbSqQ,81
+pro_craft/server/routers/admin.py,sha256=4flQtIMhVFoEf4z3rxC3W0k70Gtpupbv_JrM_AqWZnc,1608
+pro_craft/server/routers/user.py,sha256=SHNXnJDSM2hKVpKQVUnjlBN-sVWuANlrAF26TWpDCR8,723
+pro_craft/server/utils/__init__.py,sha256=ziKn9B14SbHdwnRH9cxiA-rXHuzjtPDM1mAIdRc9p-I,131
+pro_craft/server/utils/auth_backends.py,sha256=7eZ2tC7P6Tw1wBN3X5RWt1FNJKtzpZutmMg5rRbDbNQ,490
+pro_craft/server/utils/database.py,sha256=t2cTLmTT0yD9ObXZxpJrLo-d38C_QRtHTVrtLcgem3o,1083
+pro_craft/server/utils/user_manager.py,sha256=tmvmoCfblMpBQtvhYsy5YS0ANy2mSW3vvS-rsajhzVw,3380
+pro_craft-0.1.1.dist-info/METADATA,sha256=T9zezfPzqEvV39r8sEbjoIPvEx92F0iEvAM7POFnfXI,416
+pro_craft-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pro_craft-0.1.1.dist-info/top_level.txt,sha256=yqYDHArnYMWpeCxkmGRwlL6sJtxiOUnYylLDx9EOgFg,10
+pro_craft-0.1.1.dist-info/RECORD,,

pro_craft-0.1.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

pro_craft-0.1.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+pro_craft