Copy of PR #435 - make start_session non blocking

[raghavdixit99]

Enhance Client Initialization and Session Management

• Purpose:
  Improve the initialization process of the Client and enhance session management capabilities.
• Key Changes:
  • Added _initialize_autogen_logger() method to configure logging for autogen.
  • Introduced a callback mechanism in the Session class to handle session state changes.
  • Refactored session creation logic to ensure agents are created only when the session is running.
  • Updated tests to reflect changes in session handling and API key validation.
  • Modified pytest fixture scope for better isolation in tests.
• Impact:
  These changes streamline the session management process and improve logging, potentially enhancing debugging and monitoring capabilities.

✨ Generated with love by Kaizen ❤️

Original Description

## 📥 Pull Request

📘 Description
Make start session non-blocking

(This PR was copied from AgentOps-AI/agentops PR AgentOps-AI#435)

[kaizen-bot]

🔍 Code Review Summary

❗ Attention Required: This push has potential issues. 🚨

Overview

• Total Feedbacks: 2 (Critical: 2, Refinements: 0)
• Files Affected: 1
• Code Quality: [█████████████████░░░] 85% (Good)

🚨 Critical Issues

Performance (2 issues)

1. Optimize session start and stop handling

---

📁 File: agentops/session.py
🔍 Reasoning:
The current implementation of the _run method in the Session class could be optimized to improve performance and responsiveness. The method blocks the main thread while waiting for the session to start, and it also blocks the main thread when stopping the session.

💡 Solution:
Refactor the _run method to use a separate thread for the session start and stop operations. This will allow the main thread to continue processing other tasks while the session is being started or stopped.

Current Code:

["[LINE 364][UPDATED] def _run(self, callback: Optional[Callable[['Session'], None]] = None) -> None:", '[LINE 365][UPDATED]     self.is_running = self._start_session()', '[LINE 366][UPDATED] ', '[LINE 367][UPDATED]     if callback:', '[LINE 368][UPDATED]         callback(self)', '[LINE 369][UPDATED] ', '[LINE 370][UPDATED]     if self.is_running == False:', '[LINE 371][UPDATED]         self.stop_flag.set()', '[LINE 372][UPDATED]         self.thread.join(timeout=1)']

Suggested Code:

["[LINE 364] def _run(self, callback: Optional[Callable[['Session'], None]] = None) -> None:", '[LINE 365]     self.is_running = self._start_session_async(callback)', '[LINE 366] ', '[LINE 367]     while not self.stop_flag.is_set():', '[LINE 368]         time.sleep(self.config.max_wait_time / 1000)', '[LINE 369]         if self.queue and self.jwt is not None:', '[LINE 370]             self._flush_queue()', '[LINE 371] ', '[LINE 372]     self._stop_session()', '[LINE 373] ', "[LINE 374] def _start_session_async(self, callback: Optional[Callable[['Session'], None]] = None) -> bool:", '[LINE 375]     self.thread = threading.Thread(target=self._start_session_worker, args=(callback,))', '[LINE 376]     self.thread.daemon = True', '[LINE 377]     self.thread.start()', '[LINE 378]     return True', '[LINE 379] ', '[LINE 380] def _stop_session(self) -> None:', '[LINE 381]     self.stop_flag.set()', '[LINE 382]     self.thread.join(timeout=1)']

2. Ensure API key is not logged in error messages

---

📁 File: agentops/session.py
🔍 Reasoning:
The current implementation logs the API key in error messages, which could potentially expose sensitive information and compromise the security of the system.

💡 Solution:
Remove the API key from the error messages to avoid leaking sensitive information.

Current Code:

['[LINE 286][UPDATED] logger.error(f"Could not start session - server could not authenticate your API Key")']

Suggested Code:

['[LINE 286] logger.error("Could not start session - server could not authenticate your API Key")']

---

✨ Generated with love by Kaizen ❤️

Useful Commands

• Feedback: Share feedback on kaizens performance with !feedback [your message]
• Ask PR: Reply with !ask-pr [your question]
• Review: Reply with !review
• Update Tests: Reply with !unittest to create a PR with test changes