pro-craft 0.1.24__tar.gz → 0.1.25__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of pro-craft might be problematic. Click here for more details.
- {pro_craft-0.1.24 → pro_craft-0.1.25}/PKG-INFO +1 -1
- {pro_craft-0.1.24 → pro_craft-0.1.25}/pyproject.toml +1 -1
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/prompt_craft/async_.py +0 -4
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/server/router/prompt.py +11 -9
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft.egg-info/PKG-INFO +1 -1
- {pro_craft-0.1.24 → pro_craft-0.1.25}/README.md +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/setup.cfg +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/__init__.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/code_helper/coder.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/code_helper/designer.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/database.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/file_manager.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/log.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/prompt_craft/__init__.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/prompt_craft/evals.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/prompt_craft/new.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/prompt_craft/sync.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/server/mcp/__init__.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/server/mcp/prompt.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/server/router/__init__.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft/utils.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft.egg-info/SOURCES.txt +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft.egg-info/dependency_links.txt +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft.egg-info/requires.txt +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/src/pro_craft.egg-info/top_level.txt +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/tests/test22.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/tests/test_coder.py +0 -0
- {pro_craft-0.1.24 → pro_craft-0.1.25}/tests/test_designer.py +0 -0
|
@@ -1,12 +1,13 @@
|
|
|
1
1
|
|
|
2
2
|
|
|
3
3
|
from fastapi import APIRouter
|
|
4
|
-
from pro_craft import Intel
|
|
4
|
+
from pro_craft import Intel,AsyncIntel
|
|
5
5
|
from pro_craft.utils import create_session
|
|
6
6
|
|
|
7
7
|
def create_router(database_url: str,
|
|
8
8
|
slave_database_url: str,
|
|
9
|
-
model_name: str
|
|
9
|
+
model_name: str,
|
|
10
|
+
logger = None):
|
|
10
11
|
"""
|
|
11
12
|
# TODO 整理改为异步
|
|
12
13
|
创建一个包含 ProCraft 路由的 FastAPI APIRouter 实例。
|
|
@@ -21,9 +22,10 @@ def create_router(database_url: str,
|
|
|
21
22
|
APIRouter: 配置好的 FastAPI APIRouter 实例。
|
|
22
23
|
"""
|
|
23
24
|
|
|
24
|
-
intels =
|
|
25
|
+
intels = AsyncIntel(
|
|
25
26
|
database_url=database_url,
|
|
26
|
-
model_name=model_name
|
|
27
|
+
model_name=model_name,
|
|
28
|
+
logger=logger
|
|
27
29
|
)
|
|
28
30
|
|
|
29
31
|
router = APIRouter(
|
|
@@ -33,7 +35,7 @@ def create_router(database_url: str,
|
|
|
33
35
|
@router.get("/push_order",
|
|
34
36
|
description="可选 train,inference,summary,finetune,patch",)
|
|
35
37
|
async def push_order(demand: str, prompt_id: str, action_type: str = "train"):
|
|
36
|
-
result = intels.push_action_order(
|
|
38
|
+
result = await intels.push_action_order(
|
|
37
39
|
demand=demand,
|
|
38
40
|
prompt_id=prompt_id,
|
|
39
41
|
action_type=action_type
|
|
@@ -43,7 +45,7 @@ def create_router(database_url: str,
|
|
|
43
45
|
@router.get("/get_latest_prompt")
|
|
44
46
|
async def get_latest_prompt(prompt_id: str):
|
|
45
47
|
with create_session(intels.engine) as session:
|
|
46
|
-
result = intels.get_prompts_from_sql(
|
|
48
|
+
result = await intels.get_prompts_from_sql(
|
|
47
49
|
prompt_id=prompt_id,
|
|
48
50
|
session=session
|
|
49
51
|
)
|
|
@@ -51,20 +53,20 @@ def create_router(database_url: str,
|
|
|
51
53
|
|
|
52
54
|
@router.get("/sync_database")
|
|
53
55
|
async def sync_database():
|
|
54
|
-
result = intels.sync_prompt_data_to_database(slave_database_url)
|
|
56
|
+
result = await intels.sync_prompt_data_to_database(slave_database_url)
|
|
55
57
|
return {"message": "success","result":result}
|
|
56
58
|
|
|
57
59
|
|
|
58
60
|
@router.get("/roll_back")
|
|
59
61
|
async def roll_back(prompt_id:str,version:str):
|
|
60
62
|
with create_session(intels.engine) as session:
|
|
61
|
-
result = intels.get_prompts_from_sql(
|
|
63
|
+
result = await intels.get_prompts_from_sql(
|
|
62
64
|
prompt_id=prompt_id,
|
|
63
65
|
version = version,
|
|
64
66
|
session=session
|
|
65
67
|
)
|
|
66
68
|
assert result.version == version
|
|
67
|
-
intels.save_prompt_increment_version(
|
|
69
|
+
await intels.save_prompt_increment_version(
|
|
68
70
|
prompt_id = prompt_id,
|
|
69
71
|
new_prompt = result.prompt,
|
|
70
72
|
use_case = result.use_case,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|