diff --git a/agentops/worker.py b/agentops/worker.py index 9ff2cd32f..233b77f9b 100644 --- a/agentops/worker.py +++ b/agentops/worker.py @@ -19,7 +19,7 @@ def __init__(self, config: Configuration) -> None: self.thread.daemon = True self.thread.start() self._session: Optional[Session] = None - self.jwt_token = None + self.jwt = None def add_event(self, event: dict) -> None: with self.lock: @@ -41,7 +41,7 @@ def flush_queue(self) -> None: serialized_payload = safe_serialize(payload).encode("utf-8") HttpClient.post(f'{self.config.endpoint}/v2/create_events', serialized_payload, - jwt_token=self.jwt_token) + jwt_token=self.jwt) logger.debug("\n") logger.debug(f"Worker request to {self.config.endpoint}/events") @@ -65,8 +65,8 @@ def start_session(self, session: Session) -> bool: if res.code != 200: return False - self.jwt_token = res.body.get('token', None) - if self.jwt_token is None: + self.jwt = res.body.get('jwt', None) + if self.jwt is None: return False return True @@ -85,7 +85,7 @@ def end_session(self, session: Session) -> str: res = HttpClient.post(f'{self.config.endpoint}/v2/update_session', json.dumps(filter_unjsonable( payload)).encode("utf-8"), - jwt_token=self.jwt_token) + jwt_token=self.jwt) logger.debug(res.body) return res.body.get('token_cost', "unknown") @@ -98,7 +98,7 @@ def update_session(self, session: Session) -> None: res = HttpClient.post(f'{self.config.endpoint}/v2/update_session', json.dumps(filter_unjsonable( payload)).encode("utf-8"), - jwt_token=self.jwt_token) + jwt_token=self.jwt) def create_agent(self, agent_id, name): payload = { @@ -111,7 +111,7 @@ def create_agent(self, agent_id, name): safe_serialize(payload).encode("utf-8") HttpClient.post(f'{self.config.endpoint}/v2/create_agent', serialized_payload, - jwt_token=self.jwt_token) + jwt_token=self.jwt) def run(self) -> None: while not self.stop_flag.is_set():