Skip to content

Commit 5bc7714

Browse files
committed
fix conflicts
1 parent 97818bf commit 5bc7714

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/api/test_turntaking_atomic.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,6 @@
55
from typing import List, Dict, Any, Optional, Union
66
import logging
77
from pydantic import BaseModel
8-
9-
logging.basicConfig(
10-
level=logging.DEBUG,
11-
format="%(asctime)s - %(levelname)s - %(message)s",
12-
force=True,
13-
)
14-
15-
# Import required components from the Sotopia codebase.
168
from sotopia.api.websocket_utils import (
179
build_observation,
1810
get_env_agents,
@@ -30,6 +22,15 @@
3022
from sotopia.database import EnvironmentProfile, AgentProfile, EvaluationDimensionBuilder
3123
from sotopia.database.persistent_profile import RelationshipType
3224

25+
26+
logging.basicConfig(
27+
level=logging.DEBUG,
28+
format="%(asctime)s - %(levelname)s - %(message)s",
29+
force=True,
30+
)
31+
32+
# Import required components from the Sotopia codebase.
33+
3334
# =============================================================================
3435
# Dummy classes to simulate database objects
3536
# =============================================================================
@@ -206,7 +207,6 @@ def __init__(self) -> None:
206207
"content": dummy_msgs["agent1"].to_natural_language()
207208
}]
208209
async def process_turn(self, client_data: Dict[str, str]) -> Dict[str, Union[int, str]]:
209-
from sotopia.api.websocket_utils import build_observation
210210
self.conversation_history.append({"role": "client", "content": client_data.get("content", "")})
211211
agent_id: str = client_data.get("agent_id")
212212
if agent_id not in self.agents:

0 commit comments

Comments
 (0)