neuro-simulator 0.4.3__py3-none-any.whl → 0.5.0__py3-none-any.whl

{neuro_simulator-0.4.3.dist-info → neuro_simulator-0.5.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 neuro_simulator/__init__.py,sha256=-tposzyvg6UckPcfSvtc03UjxBa9oCe_zRvlKf8splk,31
 neuro_simulator/cli.py,sha256=gtmZZ7Y-lZqEtz9AIA4TuJ4vJw5pNFvPrqpxxIaDnl4,5455
-neuro_simulator/config.yaml.example,sha256=RbSHEAFD9NEuIvgl5Smpasfz4jiZYQgzTovL3MrhhIs,6549
+neuro_simulator/config.yaml.example,sha256=M-uCuDj-0XvM6BuFNQT9rsFHL_t71UK0jWVSuF6nL6Q,4732
 neuro_simulator/agent/__init__.py,sha256=t52CZlyTGWqcGjMs90qvpFpRckY2WSSlO7r_H3K_mSY,32
 neuro_simulator/agent/core.py,sha256=_UkxMV1S33VHjD1JFlKz17GpaDhHQGNA9s9dsRgd3j0,9561
 neuro_simulator/agent/llm.py,sha256=xPBEXpZ19WOt7YkERNaY9rscNI-ePASTjIt-_sZV7UI,4262
@@ -25,26 +25,38 @@ neuro_simulator/agent/tools/update_core_memory_block.py,sha256=CQgimyPNLmOuK_pDA
 neuro_simulator/api/__init__.py,sha256=5LWyDSayPGdQS8Rv13nmAKLyhPnMVPyTYDdvoMPB4xw,56
 neuro_simulator/api/system.py,sha256=OJT6m6HIYATMCAKrgeBRhficaiUjIDl9f-WyUT-RoBw,1874
 neuro_simulator/assets/neuro_start.mp4,sha256=xCLnNzv4THnzRYwkdV6EiqXc-XtFd867R2ZVLDvNp0Y,8226418
+neuro_simulator/chatbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+neuro_simulator/chatbot/core.py,sha256=jddeOZDYcEcWpEryMVOlLFCq1YqIZGnqQi7VV_sPnRU,11314
+neuro_simulator/chatbot/llm.py,sha256=UgiAq8aqmuSaf3poX_QIZHPxKPZpr_Xn4aHrhTlESPE,4204
+neuro_simulator/chatbot/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+neuro_simulator/chatbot/memory/manager.py,sha256=K84EEH-gRPksgt7GcNxvEVo60aHxaOTC3dZMf2rR9bQ,3827
+neuro_simulator/chatbot/nickname_gen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+neuro_simulator/chatbot/nickname_gen/generator.py,sha256=ajrGp37uD7kzzzl9msMeD2EWyYW2JBXE2VT3N4IKpBA,6361
+neuro_simulator/chatbot/prompts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+neuro_simulator/chatbot/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+neuro_simulator/chatbot/tools/add_temp_memory.py,sha256=ZsqvVx1cZMyjVFTo8ICxdhuk72K2kF17N9imDASODz8,1884
+neuro_simulator/chatbot/tools/base.py,sha256=YebWl1Y0uyMDzehxXWpmbQfPxGcZg1n3cJO2l13wRdc,1208
+neuro_simulator/chatbot/tools/manager.py,sha256=LJx6teTURs7uPstWYS-odyNr32AwDErxRQzqKRhxmjI,5064
+neuro_simulator/chatbot/tools/post_chat_message.py,sha256=lchGCSQKzPZgb1z__MIDkG2Cxn_nAN3ISsNkYGbpTsg,1775
 neuro_simulator/core/__init__.py,sha256=-ojq25c8XA0CU25b0OxcGjH4IWFEDHR-HXSRSZIuKe8,57
 neuro_simulator/core/agent_factory.py,sha256=p3IKT6sNzSpUojQjvbZJu0pbmyyb258sGn_YNSJlqwI,1717
 neuro_simulator/core/agent_interface.py,sha256=ZXUCtkQUvsBQ5iCb0gTILJaShn5KmSrEgKhd7PK18HE,2794
-neuro_simulator/core/application.py,sha256=mW5SlqonSn5oQvb5xuGR8FTZ_-sEx3OUTXwfC_N_-2c,26703
-neuro_simulator/core/config.py,sha256=JNcZccxRa0oY_6ctRlue_2Wf2zxL0FONr8jXVE8xqfg,15568
-neuro_simulator/core/path_manager.py,sha256=hfjI4s8-WXlgwvPWDSLYMQ2d0OaXPOMYWWlP5xe5NLg,2954
+neuro_simulator/core/application.py,sha256=TxqDz4buITapVwgtO6Pv5-JlG--8JPk5IQraaTFW8oQ,26566
+neuro_simulator/core/config.py,sha256=BszgxKFtHyoYUYUjJmY-kc-X0H7LQencJFKzt56j_hQ,15068
+neuro_simulator/core/path_manager.py,sha256=fQZ5RLdiWYeWXbmnhmgCoBEMWXtnPEGtZ6dhtpyXRCI,4339
 neuro_simulator/services/__init__.py,sha256=s3ZrAHg5TpJakadAAGY1h0wDw_xqN4Je4aJwJyRBmk4,61
-neuro_simulator/services/audience.py,sha256=sAmvkz1ip1MNqaw7t-9abqfnmp0yh8EdG5bS5KYR-Us,4999
 neuro_simulator/services/audio.py,sha256=EElBue80njZY8_CKQhVtZEcGchJUdmS4EDOvJEEu_LY,2909
 neuro_simulator/services/builtin.py,sha256=7b5F2StGX2ZT1jMiAAuqov0mx5OMHvUYheo8E1c9fjQ,6097
 neuro_simulator/services/letta.py,sha256=OEM4qYeRRPoj8qY645YZwCHmMfg1McLp0eTbSAu3KL0,11976
-neuro_simulator/services/stream.py,sha256=eueAaxhAwX_n0mT1W0W4YOVCe7fH5uPoMdRf2o0wQMI,5937
+neuro_simulator/services/stream.py,sha256=9k8bNnQu237WMf1qX8OMfItAPEZkaZ4NEiI4FQ_7Ww4,6058
 neuro_simulator/utils/__init__.py,sha256=xSEFzjT827W81mNyQ_DLtr00TgFlttqfFgpz9pSxFXQ,58
 neuro_simulator/utils/logging.py,sha256=fAM8mW4czr0RLgAUS6oN-YTWKj5daqwc5g7yjyAgaPw,3892
 neuro_simulator/utils/process.py,sha256=9OYWx8fzaJZqmFUcjQX37AnBhl7YWvrLxDWBa30vqwU,3192
-neuro_simulator/utils/queue.py,sha256=bg-mIFF8ycClwmmfcKSPjAXtvjImzTKf6z7xZc2VX20,2196
+neuro_simulator/utils/queue.py,sha256=pyBiFSSO1ocUF0kZfXvOywVcyYAwk8XDBXR8jJLkgSI,2701
 neuro_simulator/utils/state.py,sha256=DdBqSAYfjOFtJfB1hEGhYPh32r1ZvFuVlN_-29_-luA,664
 neuro_simulator/utils/websocket.py,sha256=fjC-qipzjgM_e7XImP12DFmLhvxkMOSr2GnUWws7esE,2058
-neuro_simulator-0.4.3.dist-info/METADATA,sha256=2cAnGJWSywTFI4vjaTyogCXdh6n6T681-Bk4PBSg-hs,7675
-neuro_simulator-0.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-neuro_simulator-0.4.3.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
-neuro_simulator-0.4.3.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
-neuro_simulator-0.4.3.dist-info/RECORD,,
+neuro_simulator-0.5.0.dist-info/METADATA,sha256=yaElYaysfpSen1W2igb9eR8eTZWHF2N7F-O-4w1Ig3c,7675
+neuro_simulator-0.5.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+neuro_simulator-0.5.0.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
+neuro_simulator-0.5.0.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
+neuro_simulator-0.5.0.dist-info/RECORD,,

neuro_simulator/services/audience.py

@@ -1,102 +0,0 @@
-# neuro_simulator/services/audience.py
-import logging
-
-from google import genai
-from google.genai import types
-from openai import AsyncOpenAI
-
-from ..core.config import config_manager, AppSettings
-from ..utils.state import app_state
-
-logger = logging.getLogger(__name__.replace("neuro_simulator", "server", 1))
-
-class AudienceLLMClient:
-    async def generate_chat_messages(self, prompt: str, max_tokens: int) -> str:
-        raise NotImplementedError
-
-class GeminiAudienceLLM(AudienceLLMClient):
-    def __init__(self, api_key: str, model_name: str):
-        if not api_key:
-            raise ValueError("Gemini API Key is not provided for GeminiAudienceLLM.")
-        self.client = genai.Client(api_key=api_key)
-        self.model_name = model_name
-        logger.info(f"Initialized GeminiAudienceLLM (new SDK), model: {self.model_name}")
-
-    async def generate_chat_messages(self, prompt: str, max_tokens: int) -> str:
-        response = await self.client.aio.models.generate_content(
-            model=self.model_name,
-            contents=prompt,
-            config=types.GenerateContentConfig(
-                temperature=config_manager.settings.audience_simulation.llm_temperature,
-                max_output_tokens=max_tokens
-            )
-        )
-        raw_chat_text = ""
-        if hasattr(response, 'text') and response.text:
-            raw_chat_text = response.text
-        elif response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
-            for part in response.candidates[0].content.parts:
-                if hasattr(part, 'text') and part.text:
-                    raw_chat_text += part.text
-        return raw_chat_text
-
-class OpenAIAudienceLLM(AudienceLLMClient):
-    def __init__(self, api_key: str, model_name: str, base_url: str | None):
-        if not api_key:
-            raise ValueError("OpenAI API Key is not provided for OpenAIAudienceLLM.")
-        self.client = AsyncOpenAI(api_key=api_key, base_url=base_url)
-        self.model_name = model_name
-        logger.info(f"Initialized OpenAIAudienceLLM, model: {self.model_name}, API Base: {base_url}")
-
-    async def generate_chat_messages(self, prompt: str, max_tokens: int) -> str:
-        response = await self.client.chat.completions.create(
-            model=self.model_name,
-            messages=[{"role": "user", "content": prompt}],
-            temperature=config_manager.settings.audience_simulation.llm_temperature,
-            max_tokens=max_tokens,
-        )
-        if response.choices and response.choices[0].message and response.choices[0].message.content:
-            return response.choices[0].message.content.strip()
-        return ""
-
-async def get_dynamic_audience_prompt() -> str:
-    current_neuro_speech = ""
-    async with app_state.neuro_last_speech_lock:
-        current_neuro_speech = app_state.neuro_last_speech
-
-    prompt = config_manager.settings.audience_simulation.prompt_template.format(
-        neuro_speech=current_neuro_speech,
-        num_chats_to_generate=config_manager.settings.audience_simulation.chats_per_batch
-    )
-    return prompt
-
-class AudienceChatbotManager:
-    def __init__(self):
-        self.client: AudienceLLMClient = self._create_client(config_manager.settings)
-        self._last_checked_settings: dict = config_manager.settings.audience_simulation.model_dump()
-        logger.info("AudienceChatbotManager initialized.")
-
-    def _create_client(self, settings: AppSettings) -> AudienceLLMClient:
-        provider = settings.audience_simulation.llm_provider
-        logger.info(f"Creating new audience LLM client for provider: {provider}")
-        if provider.lower() == "gemini":
-            if not settings.api_keys.gemini_api_key:
-                raise ValueError("GEMINI_API_KEY not set in config")
-            return GeminiAudienceLLM(api_key=settings.api_keys.gemini_api_key, model_name=settings.audience_simulation.gemini_model)
-        elif provider.lower() == "openai":
-            if not settings.api_keys.openai_api_key:
-                raise ValueError("OPENAI_API_KEY not set in config")
-            return OpenAIAudienceLLM(api_key=settings.api_keys.openai_api_key, model_name=settings.audience_simulation.openai_model, base_url=settings.api_keys.openai_api_base_url)
-        else:
-            raise ValueError(f"Unsupported AUDIENCE_LLM_PROVIDER: {provider}")
-
-    def handle_config_update(self, new_settings: AppSettings):
-        new_audience_settings = new_settings.audience_simulation.model_dump()
-        if new_audience_settings != self._last_checked_settings:
-            logger.info("Audience simulation settings changed, re-initializing LLM client...")
-            try:
-                self.client = self._create_client(new_settings)
-                self._last_checked_settings = new_audience_settings
-                logger.info("LLM client hot-reloaded successfully.")
-            except Exception as e:
-                logger.error(f"Error hot-reloading LLM client: {e}", exc_info=True)