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))