Spaces:
Sleeping
Sleeping
Update agents_config.py
Browse files- agents_config.py +31 -20
agents_config.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import autogen
|
| 2 |
|
| 3 |
-
# ال
|
| 4 |
MODEL_MAP = {
|
| 5 |
"Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
|
| 6 |
"StyleGuardian": "openai/gpt-oss-120b",
|
|
@@ -14,42 +14,53 @@ MODEL_MAP = {
|
|
| 14 |
}
|
| 15 |
|
| 16 |
def get_llm_config(model_id, user_token):
|
|
|
|
| 17 |
return {
|
| 18 |
-
"config_list": [
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
|
|
|
|
|
|
| 24 |
"temperature": 0.7,
|
| 25 |
-
"cache_seed": None,
|
| 26 |
}
|
| 27 |
|
| 28 |
def create_agents(user_token):
|
| 29 |
-
|
|
|
|
|
|
|
|
|
|
| 30 |
return autogen.AssistantAgent(
|
| 31 |
name=name,
|
| 32 |
-
system_message=
|
| 33 |
llm_config=get_llm_config(MODEL_MAP[model_key], user_token)
|
| 34 |
)
|
| 35 |
|
| 36 |
agents = {
|
| 37 |
-
"analyst": mk_asst("Analyst", "محلل ن
|
| 38 |
-
"style_guardian": mk_asst("Style_Guardian", "
|
| 39 |
-
"architect": mk_asst("Architect", "مهندس حبكة.", "Architect"),
|
| 40 |
-
"draft_writer": mk_asst("Draft_Writer", "كاتب مسودة.", "DraftWriter"),
|
| 41 |
-
"humanizer": mk_asst("Humanizer", "خبير مشاعر.", "Humanizer"),
|
| 42 |
-
"continuity_guard": mk_asst("Continuity_Guard", "مراقب استمرارية.", "ContinuityGuard"),
|
| 43 |
-
"psychologist": mk_asst("Psychologist", "محلل نفسي.", "Psychologist"),
|
| 44 |
-
"critic": mk_asst("Critic", "ناقد أدبي.", "Critic"),
|
| 45 |
}
|
| 46 |
|
|
|
|
|
|
|
| 47 |
agents["editor"] = autogen.UserProxyAgent(
|
| 48 |
name="Editor_In_Chief",
|
| 49 |
-
human_input_mode="NEVER",
|
| 50 |
max_consecutive_auto_reply=10,
|
| 51 |
code_execution_config=False,
|
| 52 |
llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
|
| 53 |
-
system_message="أنت رئيس التحرير
|
|
|
|
|
|
|
| 54 |
)
|
|
|
|
| 55 |
return agents
|
|
|
|
| 1 |
import autogen
|
| 2 |
|
| 3 |
+
# خريطة النماذج المباشرة الموزعة على الوكلاء
|
| 4 |
MODEL_MAP = {
|
| 5 |
"Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
|
| 6 |
"StyleGuardian": "openai/gpt-oss-120b",
|
|
|
|
| 14 |
}
|
| 15 |
|
| 16 |
def get_llm_config(model_id, user_token):
|
| 17 |
+
"""إنشاء إعدادات الاتصال بالنموذج عبر راوتر Hugging Face"""
|
| 18 |
return {
|
| 19 |
+
"config_list": [
|
| 20 |
+
{
|
| 21 |
+
"model": model_id,
|
| 22 |
+
"api_key": user_token,
|
| 23 |
+
"base_url": "https://router.huggingface.co/v1",
|
| 24 |
+
"api_type": "openai",
|
| 25 |
+
}
|
| 26 |
+
],
|
| 27 |
"temperature": 0.7,
|
| 28 |
+
"cache_seed": None, # لضمان نتائج متجددة دائماً
|
| 29 |
}
|
| 30 |
|
| 31 |
def create_agents(user_token):
|
| 32 |
+
"""توليد فريق الوكلاء وتوزيع المهام عليهم"""
|
| 33 |
+
|
| 34 |
+
# وظيفة داخلية لتبسيط إنشاء الوكلاء المساعدين
|
| 35 |
+
def mk_asst(name, sys_msg, model_key):
|
| 36 |
return autogen.AssistantAgent(
|
| 37 |
name=name,
|
| 38 |
+
system_message=sys_msg,
|
| 39 |
llm_config=get_llm_config(MODEL_MAP[model_key], user_token)
|
| 40 |
)
|
| 41 |
|
| 42 |
agents = {
|
| 43 |
+
"analyst": mk_asst("Analyst", "أنت المحلل البنيوي للنص، فكك الأفكار وحدد الروابط العمقية.", "Analyst"),
|
| 44 |
+
"style_guardian": mk_asst("Style_Guardian", "أنت حارس الأسلوب الأدبي، تأكد من جودة اللغة والجماليات السردية.", "StyleGuardian"),
|
| 45 |
+
"architect": mk_asst("Architect", "أنت مهندس الحبكة، قم ببناء تسلسل أحداث متماسك ومشوق.", "Architect"),
|
| 46 |
+
"draft_writer": mk_asst("Draft_Writer", "أنت كاتب المسودة الأولية، حول الأفكار إلى نص سردي روائي.", "DraftWriter"),
|
| 47 |
+
"humanizer": mk_asst("Humanizer", "أنت خبير الأنسنة، بث المشاعر في الشخصيات واجعل الحوارات نابضة بالحياة.", "Humanizer"),
|
| 48 |
+
"continuity_guard": mk_asst("Continuity_Guard", "أنت مراقب الاستمرارية، امنع التناقضات في الأحداث أو صفات الشخصيات.", "ContinuityGuard"),
|
| 49 |
+
"psychologist": mk_asst("Psychologist", "أنت محلل الدوافع النفسية، اجعل تصرفات الشخصيات مبررة نفسياً.", "Psychologist"),
|
| 50 |
+
"critic": mk_asst("Critic", "أنت الناقد الأدبي، راجع النص النهائي وقدم ملاحظات قاسية للتحسين.", "Critic"),
|
| 51 |
}
|
| 52 |
|
| 53 |
+
# رئيس التحرير: المسؤول عن الإدارة والصياغة النهائية
|
| 54 |
+
# تم ضبط human_input_mode إلى NEVER لضمان العمل التلقائي في بيئة الويب
|
| 55 |
agents["editor"] = autogen.UserProxyAgent(
|
| 56 |
name="Editor_In_Chief",
|
| 57 |
+
human_input_mode="NEVER",
|
| 58 |
max_consecutive_auto_reply=10,
|
| 59 |
code_execution_config=False,
|
| 60 |
llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
|
| 61 |
+
system_message="""أنت رئيس التحرير (Ling-1T).
|
| 62 |
+
مهمتك هي قيادة نقاش الوكلاء ودمج مخرجاتهم لصياغة الفصل الروائي النهائي بأسلوب أدبي رفيع.
|
| 63 |
+
يجب أن تنتهي المحادثة دائماً بالنص الكامل للمخطوطة."""
|
| 64 |
)
|
| 65 |
+
|
| 66 |
return agents
|