agents_config.py

import autogen

# النماذج المحددة من قبلك
MODEL_MAP = {
    "Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
    "StyleGuardian": "openai/gpt-oss-120b", 
    "Architect": "deepseek-ai/DeepSeek-V3.2-Exp",
    "DraftWriter": "deepcogito/cogito-v2-preview-llama-405B",
    "Humanizer": "deepcogito/cogito-671b-v2.1",
    "ContinuityGuard": "meta-llama/Llama-3.3-70B-Instruct",
    "Psychologist": "zai-org/GLM-4.7-FP8",
    "Critic": "Qwen/Qwen3-235B-A22B-Thinking-2507",
    "Editor": "InclusionAI/Ling-1T"
}

def get_llm_config(model_id, user_token):
    return {
        "config_list": [
            {
                "model": model_id,
                "api_key": user_token,
                "base_url": f"https://api-inference.huggingface.co/models/{model_id}/v1",
                "api_type": "openai", 
            }
        ],
        "temperature": 0.7,
        "cache_seed": None,
    }

def create_agents(user_token):
    """
    هذه هي الدالة التي يبحث عنها ملف app.py
    """
    # إنشاء الوكلاء
    analyst = autogen.AssistantAgent(
        name="Analyst",
        system_message="أنت المحلل. استخرج سياق الرواية والأحداث السابقة.",
        llm_config=get_llm_config(MODEL_MAP["Analyst"], user_token)
    )

    style_guardian = autogen.AssistantAgent(
        name="Style_Guardian",
        system_message="أنت حارس الأسلوب. حافظ على أسلوب الكاتب الأصلي.",
        llm_config=get_llm_config(MODEL_MAP["StyleGuardian"], user_token)
    )

    architect = autogen.AssistantAgent(
        name="Architect",
        system_message="أنت المهندس السردي. نظم الحبكة واقترح المسارات.",
        llm_config=get_llm_config(MODEL_MAP["Architect"], user_token)
    )

    draft_writer = autogen.AssistantAgent(
        name="Draft_Writer",
        system_message="أنت كاتب الإكمال. اكتب الأجزاء المفقودة.",
        llm_config=get_llm_config(MODEL_MAP["DraftWriter"], user_token)
    )

    humanizer = autogen.AssistantAgent(
        name="Humanizer",
        system_message="أنت مضفي الطابع البشري. أضف الإبداع والحوار الطبيعي.",
        llm_config=get_llm_config(MODEL_MAP["Humanizer"], user_token)
    )

    continuity_guard = autogen.AssistantAgent(
        name="Continuity_Guard",
        system_message="أنت حارس الاستمرارية. تأكد من التناسق الزمني.",
        llm_config=get_llm_config(MODEL_MAP["ContinuityGuard"], user_token)
    )

    psychologist = autogen.AssistantAgent(
        name="Psychologist",
        system_message="أنت المدقق النفسي. حلل سلوك الشخصيات.",
        llm_config=get_llm_config(MODEL_MAP["Psychologist"], user_token)
    )

    critic = autogen.AssistantAgent(
        name="Critic",
        system_message="أنت الناقد القاسي. قدم النقد النهائي.",
        llm_config=get_llm_config(MODEL_MAP["Critic"], user_token)
    )

    editor = autogen.UserProxyAgent(
        name="Editor_In_Chief",
        human_input_mode="NEVER",
        max_consecutive_auto_reply=10,
        code_execution_config=False,
        llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
        system_message="أنت رئيس التحرير (InclusionAI/Ling-1T). وظيفتك دمج المخرجات في نص واحد متسق."
    )

    return {
        "analyst": analyst,
        "style_guardian": style_guardian,
        "architect": architect,
        "draft_writer": draft_writer,
        "humanizer": humanizer,
        "continuity_guard": continuity_guard,
        "psychologist": psychologist,
        "critic": critic,
        "editor": editor
    }