Spaces:
Sleeping
Sleeping
Update agents_config.py
Browse files- agents_config.py +77 -5
agents_config.py
CHANGED
|
@@ -1,16 +1,16 @@
|
|
| 1 |
import autogen
|
| 2 |
|
| 3 |
-
# ال
|
| 4 |
MODEL_MAP = {
|
| 5 |
"Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
|
| 6 |
-
"StyleGuardian": "openai/gpt-oss-120b",
|
| 7 |
"Architect": "deepseek-ai/DeepSeek-V3.2-Exp",
|
| 8 |
"DraftWriter": "deepcogito/cogito-v2-preview-llama-405B",
|
| 9 |
"Humanizer": "deepcogito/cogito-671b-v2.1",
|
| 10 |
"ContinuityGuard": "meta-llama/Llama-3.3-70B-Instruct",
|
| 11 |
"Psychologist": "zai-org/GLM-4.7-FP8",
|
| 12 |
"Critic": "Qwen/Qwen3-235B-A22B-Thinking-2507",
|
| 13 |
-
"Editor": "InclusionAI/Ling-1T"
|
| 14 |
}
|
| 15 |
|
| 16 |
def get_llm_config(model_id, user_token):
|
|
@@ -24,7 +24,79 @@ def get_llm_config(model_id, user_token):
|
|
| 24 |
}
|
| 25 |
],
|
| 26 |
"temperature": 0.7,
|
| 27 |
-
"
|
| 28 |
}
|
| 29 |
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import autogen
|
| 2 |
|
| 3 |
+
# النماذج المحددة من قبلك
|
| 4 |
MODEL_MAP = {
|
| 5 |
"Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
|
| 6 |
+
"StyleGuardian": "openai/gpt-oss-120b",
|
| 7 |
"Architect": "deepseek-ai/DeepSeek-V3.2-Exp",
|
| 8 |
"DraftWriter": "deepcogito/cogito-v2-preview-llama-405B",
|
| 9 |
"Humanizer": "deepcogito/cogito-671b-v2.1",
|
| 10 |
"ContinuityGuard": "meta-llama/Llama-3.3-70B-Instruct",
|
| 11 |
"Psychologist": "zai-org/GLM-4.7-FP8",
|
| 12 |
"Critic": "Qwen/Qwen3-235B-A22B-Thinking-2507",
|
| 13 |
+
"Editor": "InclusionAI/Ling-1T"
|
| 14 |
}
|
| 15 |
|
| 16 |
def get_llm_config(model_id, user_token):
|
|
|
|
| 24 |
}
|
| 25 |
],
|
| 26 |
"temperature": 0.7,
|
| 27 |
+
"cache_seed": None,
|
| 28 |
}
|
| 29 |
|
| 30 |
+
def create_agents(user_token):
|
| 31 |
+
"""
|
| 32 |
+
هذه هي الدالة التي يبحث عنها ملف app.py
|
| 33 |
+
"""
|
| 34 |
+
# إنشاء الوكلاء
|
| 35 |
+
analyst = autogen.AssistantAgent(
|
| 36 |
+
name="Analyst",
|
| 37 |
+
system_message="أنت المحلل. استخرج سياق الرواية والأحداث السابقة.",
|
| 38 |
+
llm_config=get_llm_config(MODEL_MAP["Analyst"], user_token)
|
| 39 |
+
)
|
| 40 |
+
|
| 41 |
+
style_guardian = autogen.AssistantAgent(
|
| 42 |
+
name="Style_Guardian",
|
| 43 |
+
system_message="أنت حارس الأسلوب. حافظ على أسلوب الكاتب الأصلي.",
|
| 44 |
+
llm_config=get_llm_config(MODEL_MAP["StyleGuardian"], user_token)
|
| 45 |
+
)
|
| 46 |
+
|
| 47 |
+
architect = autogen.AssistantAgent(
|
| 48 |
+
name="Architect",
|
| 49 |
+
system_message="أنت المهندس السردي. نظم الحبكة واقترح المسارات.",
|
| 50 |
+
llm_config=get_llm_config(MODEL_MAP["Architect"], user_token)
|
| 51 |
+
)
|
| 52 |
+
|
| 53 |
+
draft_writer = autogen.AssistantAgent(
|
| 54 |
+
name="Draft_Writer",
|
| 55 |
+
system_message="أنت كاتب الإكمال. اكتب الأجزاء المفقودة.",
|
| 56 |
+
llm_config=get_llm_config(MODEL_MAP["DraftWriter"], user_token)
|
| 57 |
+
)
|
| 58 |
+
|
| 59 |
+
humanizer = autogen.AssistantAgent(
|
| 60 |
+
name="Humanizer",
|
| 61 |
+
system_message="أنت مضفي الطابع البشري. أضف الإبداع والحوار الطبيعي.",
|
| 62 |
+
llm_config=get_llm_config(MODEL_MAP["Humanizer"], user_token)
|
| 63 |
+
)
|
| 64 |
+
|
| 65 |
+
continuity_guard = autogen.AssistantAgent(
|
| 66 |
+
name="Continuity_Guard",
|
| 67 |
+
system_message="أنت حارس الاستمرارية. تأكد من التناسق الزمني.",
|
| 68 |
+
llm_config=get_llm_config(MODEL_MAP["ContinuityGuard"], user_token)
|
| 69 |
+
)
|
| 70 |
+
|
| 71 |
+
psychologist = autogen.AssistantAgent(
|
| 72 |
+
name="Psychologist",
|
| 73 |
+
system_message="أنت المدقق النفسي. حلل سلوك الشخصيات.",
|
| 74 |
+
llm_config=get_llm_config(MODEL_MAP["Psychologist"], user_token)
|
| 75 |
+
)
|
| 76 |
+
|
| 77 |
+
critic = autogen.AssistantAgent(
|
| 78 |
+
name="Critic",
|
| 79 |
+
system_message="أنت الناقد القاسي. قدم النقد النهائي.",
|
| 80 |
+
llm_config=get_llm_config(MODEL_MAP["Critic"], user_token)
|
| 81 |
+
)
|
| 82 |
+
|
| 83 |
+
editor = autogen.UserProxyAgent(
|
| 84 |
+
name="Editor_In_Chief",
|
| 85 |
+
human_input_mode="NEVER",
|
| 86 |
+
max_consecutive_auto_reply=10,
|
| 87 |
+
code_execution_config=False,
|
| 88 |
+
llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
|
| 89 |
+
system_message="أنت رئيس التحرير (InclusionAI/Ling-1T). وظيفتك دمج المخرجات في نص واحد متسق."
|
| 90 |
+
)
|
| 91 |
+
|
| 92 |
+
return {
|
| 93 |
+
"analyst": analyst,
|
| 94 |
+
"style_guardian": style_guardian,
|
| 95 |
+
"architect": architect,
|
| 96 |
+
"draft_writer": draft_writer,
|
| 97 |
+
"humanizer": humanizer,
|
| 98 |
+
"continuity_guard": continuity_guard,
|
| 99 |
+
"psychologist": psychologist,
|
| 100 |
+
"critic": critic,
|
| 101 |
+
"editor": editor
|
| 102 |
+
}
|