GoodStartLabs/AI_Diplomacy
1
0
Fork 0
mirror of https://github.com/GoodStartLabs/AI_Diplomacy.git synced 2026-04-30 17:40:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Support for o3-pro with openai responses api

Browse source
This commit is contained in:
AlxAI 2025-06-14 17:28:01 -04:00
parent fbd92d91ba
commit 77e7921b9c
2 changed files with 113 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
ai_diplomacy/utils.py
View file

@ -34,20 +34,19 @@ def assign_models_to_powers() -> Dict[str, str]:
"""
# POWER MODELS
"""
return {
"AUSTRIA": "o3",
"ENGLAND": "gpt-4.1-2025-04-14",
"FRANCE": "o4-mini",
"FRANCE": "o3-pro",
"GERMANY": "gpt-4o",
"ITALY": "gpt-4.1-2025-04-14",
"ITALY": "o3",
"RUSSIA": "gpt-4o",
"TURKEY": "o4-mini",
}
"""
# TEST MODELS
"""
return {
"AUSTRIA": "openrouter-google/gemini-2.5-flash-preview",
"ENGLAND": "openrouter-google/gemini-2.5-flash-preview",
@ -57,6 +56,7 @@ def assign_models_to_powers() -> Dict[str, str]:
"RUSSIA": "openrouter-google/gemini-2.5-flash-preview",
"TURKEY": "openrouter-google/gemini-2.5-flash-preview",
}
"""
def gather_possible_orders(game: Game, power_name: str) -> Dict[str, List[str]]: