40 lines
1.9 KiB
Python
40 lines
1.9 KiB
Python
|
import asyncio
|
|||
|
|
import httpx
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def create_hermes_profiles(characters: list[dict], base_url: str = "http://localhost:11434/v1"):
|
|||
|
|
async with httpx.AsyncClient(timeout=60.0) as client:
|
|||
|
|
for char in characters:
|
|||
|
|
profile_name = char["name"].lower().replace(" ", "_")
|
|||
|
|
print(f"Creating profile: {profile_name}")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = await client.post(
|
|||
|
|
f"{base_url}/models",
|
|||
|
|
json={
|
|||
|
|
"name": profile_name,
|
|||
|
|
"system_prompt": f"你是{char['name']}，{char.get('personality', '')}，{char.get('background', '')}",
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
print(f" Created: {profile_name}")
|
|||
|
|
else:
|
|||
|
|
print(f" Warning ({response.status_code}): {profile_name} - {response.text[:200]}")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f" Error creating {profile_name}: {e}")
|
|||
|
|
|
|||
|
|
print("Profile creation completed.")
|
|||
|
|
|
|||
|
|
|
|||
|
|
SAMPLE_CHARACTERS = [
|
|||
|
|
{"name": "侦探_张", "personality": "冷静理智，善于观察和分析", "background": "著名私家侦探"},
|
|||
|
|
{"name": "富商_李", "personality": "精明世故，看透人心", "background": "本地富商，经营古董店"},
|
|||
|
|
{"name": "医生_王", "personality": "温柔体贴，善于倾听", "background": "在本地开诊所的医生"},
|
|||
|
|
{"name": "记者_赵", "personality": "活泼开朗，和所有人关系都很好", "background": "专门报道离奇案件的记者"},
|
|||
|
|
{"name": "艺术家_陈", "personality": "阴郁深沉，似乎藏着秘密", "background": "性格独特的艺术家"},
|
|||
|
|
{"name": "教师_林", "personality": "急躁易怒，说话带刺", "background": "学校老师"},
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
asyncio.run(create_hermes_profiles(SAMPLE_CHARACTERS))
|