Skip to content

Commit 0232f97

Browse files
committed
Allow Game Engine to be selected
Remove task description from reply template.
1 parent bb5636b commit 0232f97

2 files changed

Lines changed: 19 additions & 12 deletions

File tree

src/game_sdk/hosted_game/agent.py

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,6 @@ class ContentLLMTemplate:
164164
top_p: float = 0.7
165165
repetition_penalty: float = 1.0
166166
type: str = None
167-
taskDescription: str = None
168167
isSandbox: bool = False
169168

170169
def _validate_fields(self):
@@ -190,11 +189,7 @@ def _validate_fields(self):
190189
self.type = self.template_type
191190
self.isSandbox = False
192191
self.userPrompt = self.user_prompt or ""
193-
194-
# Additional settings for REPLY only
195-
if self.template_type == "REPLY":
196-
self.taskDescription = self.taskDescription or "Process incoming tweet. Ignore if it is boring or unimportant. Ignore if the conversation has gone too long."
197-
192+
198193
elif self.template_type in ["TWITTER_START_SYSTEM_PROMPT", "TWITTER_END_SYSTEM_PROMPT", "SHARED"]:
199194
if not self.system_prompt:
200195
raise ValueError("system_prompt is required")
@@ -259,7 +254,8 @@ def __init__(
259254
world_info: str = "",
260255
main_heartbeat: int = 15,
261256
reaction_heartbeat: int = 5,
262-
task_description: str = ""
257+
task_description: str = "",
258+
game_engine_model: str = "llama_3_1_405b"
263259
):
264260
self.game_sdk = sdk.GameSDK(api_key)
265261
self.goal = goal
@@ -272,7 +268,7 @@ def __init__(
272268
self.templates: List[ContentLLMTemplate] = []
273269
self.tweet_usernames: List[str] = []
274270
self.task_description: str = task_description
275-
271+
self.game_engine_model: str = game_engine_model
276272

277273
def set_goal(self, goal: str):
278274
self.goal = goal
@@ -298,8 +294,16 @@ def set_task_description(self, task_description: str):
298294
self.task_description = task_description
299295
return True
300296

297+
def set_game_engine_model(self, game_engine_model: str):
298+
# Available models: llama_3_1_405b, deepseek_r1, llama_3_3_70b_instruct, qwen2p5_72b_instruct, deepseek_v3
299+
self.game_engine_model = game_engine_model
300+
return True
301+
301302
def get_task_description(self) -> str:
302303
return self.task_description
304+
305+
def get_game_engine_model(self) -> str:
306+
return self.game_engine_model
303307

304308
def get_goal(self) -> str:
305309
return self.goal
@@ -378,7 +382,10 @@ def deploy_twitter(self):
378382
self.enabled_functions,
379383
self.custom_functions,
380384
self.main_heartbeat,
381-
self.reaction_heartbeat
385+
self.reaction_heartbeat,
386+
self.tweet_usernames,
387+
self.templates,
388+
self.game_engine_model
382389
)
383390

384391
def export(self) -> str:

src/game_sdk/hosted_game/sdk.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def react(self, session_id: str, platform: str, goal: str,
9090

9191
return response.json()["data"]
9292

93-
def deploy(self, goal: str, description: str, world_info: str, functions: list, custom_functions: list, main_heartbeat: int, reaction_heartbeat: int, tweet_usernames: list = None, templates: list = None, task_description: str = None):
93+
def deploy(self, goal: str, description: str, world_info: str, functions: list, custom_functions: list, main_heartbeat: int, reaction_heartbeat: int, tweet_usernames: list = None, templates: list = None, game_engine_model: str = "llama_3_1_405b"):
9494
"""
9595
Simulate the agent configuration
9696
"""
@@ -104,9 +104,9 @@ def deploy(self, goal: str, description: str, world_info: str, functions: list,
104104
"mainHeartbeat": main_heartbeat,
105105
"reactionHeartbeat": reaction_heartbeat,
106106
},
107-
"taskDescription": task_description
107+
"gameEngineModel": game_engine_model
108108
}
109-
109+
110110
if tweet_usernames is not None:
111111
payload["tweetUsernames"] = tweet_usernames
112112

0 commit comments

Comments
 (0)