Spaces:
Sleeping
Sleeping
Update agents_config.py
Browse files- agents_config.py +33 -84
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):
|
|
@@ -28,84 +28,33 @@ def get_llm_config(model_id, user_token):
|
|
| 28 |
}
|
| 29 |
|
| 30 |
def create_agents(user_token):
|
| 31 |
-
#
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
llm_config=get_llm_config(MODEL_MAP["Architect"], user_token)
|
| 50 |
-
)
|
| 51 |
-
|
| 52 |
-
# 4. كاتب المسودة
|
| 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 |
-
# 5. وكيل الأنسنة
|
| 60 |
-
humanizer = autogen.AssistantAgent(
|
| 61 |
-
name="Humanizer",
|
| 62 |
-
system_message="أنت وكيل الأنسنة. أضف تفاصيل حسية، مشاعر عميقة، وحوارات واقعية تجعل النص ينبض بالحياة.",
|
| 63 |
-
llm_config=get_llm_config(MODEL_MAP["Humanizer"], user_token)
|
| 64 |
-
)
|
| 65 |
-
|
| 66 |
-
# 6. حارس الاستمرارية
|
| 67 |
-
continuity_guard = autogen.AssistantAgent(
|
| 68 |
-
name="Continuity_Guard",
|
| 69 |
-
system_message="أنت حارس الاستمرارية. تأكد من عدم وجود تناقضات مع الأحداث الواردة في الملف الأصلي.",
|
| 70 |
-
llm_config=get_llm_config(MODEL_MAP["ContinuityGuard"], user_token)
|
| 71 |
-
)
|
| 72 |
-
|
| 73 |
-
# 7. الأخصائي النفسي
|
| 74 |
-
psychologist = autogen.AssistantAgent(
|
| 75 |
-
name="Psychologist",
|
| 76 |
-
system_message="أنت الأخصائي النفسي. تأكد من أن تصرفات الشخصيات في الإكمال تتوافق مع تركيبتهم النفسية.",
|
| 77 |
-
llm_config=get_llm_config(MODEL_MAP["Psychologist"], user_token)
|
| 78 |
-
)
|
| 79 |
-
|
| 80 |
-
# 8. الناقد
|
| 81 |
-
critic = autogen.AssistantAgent(
|
| 82 |
-
name="Critic",
|
| 83 |
-
system_message="أنت الناقد. راجع المخرج النهائي بصرامة واقترح تعديلات لتحسين الجودة الأدبية.",
|
| 84 |
-
llm_config=get_llm_config(MODEL_MAP["Critic"], user_token)
|
| 85 |
-
)
|
| 86 |
|
| 87 |
-
#
|
| 88 |
-
editor = autogen.UserProxyAgent(
|
| 89 |
name="Editor_In_Chief",
|
| 90 |
-
human_input_mode="NEVER",
|
| 91 |
max_consecutive_auto_reply=15,
|
| 92 |
code_execution_config=False,
|
| 93 |
llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
|
| 94 |
-
system_message="
|
| 95 |
-
1. استلام النص من الملف المرفوع وتوزيعه على الوكلاء.
|
| 96 |
-
2. جمع المخرجات وصهرها في 'مخطوطة نهائية' منظمة.
|
| 97 |
-
3. يجب أن يكون مخرجك النهائي منسقاً كالتالي: [العنوان] -> [رقم الفصل] -> [المحتوى الروائي].
|
| 98 |
-
أنت المسؤول عن الجودة النهائية للملف الذي سيتم تحميله."""
|
| 99 |
)
|
| 100 |
|
| 101 |
-
return
|
| 102 |
-
"analyst": analyst,
|
| 103 |
-
"style_guardian": style_guardian,
|
| 104 |
-
"architect": architect,
|
| 105 |
-
"draft_writer": draft_writer,
|
| 106 |
-
"humanizer": humanizer,
|
| 107 |
-
"continuity_guard": continuity_guard,
|
| 108 |
-
"psychologist": psychologist,
|
| 109 |
-
"critic": critic,
|
| 110 |
-
"editor": editor
|
| 111 |
-
}
|
|
|
|
| 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):
|
|
|
|
| 28 |
}
|
| 29 |
|
| 30 |
def create_agents(user_token):
|
| 31 |
+
# إنشاء الوكلاء المساعدين
|
| 32 |
+
def create_asst(name, msg, model_key):
|
| 33 |
+
return autogen.AssistantAgent(
|
| 34 |
+
name=name,
|
| 35 |
+
system_message=msg,
|
| 36 |
+
llm_config=get_llm_config(MODEL_MAP[model_key], user_token)
|
| 37 |
+
)
|
| 38 |
+
|
| 39 |
+
agents = {
|
| 40 |
+
"analyst": create_asst("Analyst", "أنت المحلل البنيوي للنص.", "Analyst"),
|
| 41 |
+
"style_guardian": create_asst("Style_Guardian", "أنت حارس الأسلوب الأدبي.", "StyleGuardian"),
|
| 42 |
+
"architect": create_asst("Architect", "أنت مهندس الحبكة السردية.", "Architect"),
|
| 43 |
+
"draft_writer": create_asst("Draft_Writer", "أنت كاتب المسودة الأولية.", "DraftWriter"),
|
| 44 |
+
"humanizer": create_asst("Humanizer", "أنت خبير الأنسنة والتفاصيل الحسية.", "Humanizer"),
|
| 45 |
+
"continuity_guard": create_asst("Continuity_Guard", "أنت مراقب الاستمرارية ومنطق الأحداث.", "ContinuityGuard"),
|
| 46 |
+
"psychologist": create_asst("Psychologist", "أنت محلل الدوافع النفسية للشخصيات.", "Psychologist"),
|
| 47 |
+
"critic": create_asst("Critic", "أنت الناقد الأدبي الصارم.", "Critic"),
|
| 48 |
+
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
|
| 50 |
+
# رئيس التحرير بوضع NEVER للعمل مع واجهة الويب
|
| 51 |
+
agents["editor"] = autogen.UserProxyAgent(
|
| 52 |
name="Editor_In_Chief",
|
| 53 |
+
human_input_mode="NEVER",
|
| 54 |
max_consecutive_auto_reply=15,
|
| 55 |
code_execution_config=False,
|
| 56 |
llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
|
| 57 |
+
system_message="أنت رئيس التحرير المسؤول عن دمج كل مساهمات الوكلاء في نص روائي واحد متكامل."
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
)
|
| 59 |
|
| 60 |
+
return agents
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|