diff --git a/agentops/client.py b/agentops/client.py index 43755272..e41fa99b 100644 --- a/agentops/client.py +++ b/agentops/client.py @@ -238,7 +238,7 @@ def start_session(self, tags: Optional[List[str]] = None, config: Optional[Confi if not config and not self.config: return logger.warning("🖇 AgentOps: Cannot start session - missing configuration") - self._session = Session(inherited_session_id or uuid4(), tags or self._tags, host_env=(get_host_env() if self._env_data_opt_out else None)) + self._session = Session(inherited_session_id or uuid4(), tags or self._tags, host_env=get_host_env(self._env_data_opt_out)) self._worker = Worker(config or self.config) start_session_result = self._worker.start_session(self._session) if not start_session_result: diff --git a/agentops/host_env.py b/agentops/host_env.py index 42974a43..3b289033 100644 --- a/agentops/host_env.py +++ b/agentops/host_env.py @@ -54,11 +54,17 @@ def get_disk_details(): return disk_info -def get_host_env(): - return { - "SDK": get_sdk_details(), - "OS": get_os_details(), - "CPU": get_cpu_details(), - "RAM": get_ram_details(), - "Disk": get_disk_details(), - } +def get_host_env(opt_out: bool = False): + if opt_out: + return { + "SDK": get_sdk_details(), + "OS": get_os_details() + } + else: + return { + "SDK": get_sdk_details(), + "OS": get_os_details(), + "CPU": get_cpu_details(), + "RAM": get_ram_details(), + "Disk": get_disk_details(), + }