Skip to content

Commit c2b4225

Browse files
committed
cleanup some namings
1 parent 809c99a commit c2b4225

3 files changed

Lines changed: 6 additions & 6 deletions

File tree

examples/game/test_agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,7 @@ def throw_furniture(object: str, **kwargs) -> Tuple[FunctionResultStatus, str, d
275275
agent_description="You are a mischievous master of chaos is very strong but with a very short attention span, and not so much brains",
276276
get_agent_state_fn=get_agent_state_fn,
277277
workers=[fruit_thrower, furniture_thrower],
278-
llm_name="Llama-3.1-405B-Instruct"
278+
model_name="Llama-3.1-405B-Instruct"
279279
)
280280

281281
# # interact and instruct the worker to do something

src/game_sdk/game/agent.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def __init__(self,
105105
agent_description: str,
106106
get_agent_state_fn: Callable,
107107
workers: Optional[List[WorkerConfig]] = None,
108-
llm_name: str = "Llama-3.1-405B-Instruct",
108+
model_name: str = "Llama-3.1-405B-Instruct",
109109
):
110110

111111
if api_key.startswith("apt-"):
@@ -115,7 +115,7 @@ def __init__(self,
115115

116116
self._api_key: str = api_key
117117

118-
self._llm_name: str = llm_name
118+
self._model_name: str = model_name
119119

120120
# checks
121121
if not self._api_key:
@@ -241,7 +241,7 @@ def _get_action(
241241
response = self.client.get_agent_action(
242242
agent_id=self.agent_id,
243243
data=data,
244-
model_name=self._llm_name
244+
model_name=self._model_name
245245
)
246246

247247
return ActionResponse.model_validate(response)

src/game_sdk/game/worker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def __init__(
5959

6060
self._api_key: str = api_key
6161

62-
self._llm_name: str = model_name
62+
self._model_name: str = model_name
6363

6464
# checks
6565
if not self._api_key:
@@ -160,7 +160,7 @@ def _get_action(
160160
self._agent_id,
161161
self._submission_id,
162162
data,
163-
model_name=self._llm_name
163+
model_name=self._model_name
164164
)
165165

166166
return ActionResponse.model_validate(response)

0 commit comments

Comments
 (0)