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

neuro_simulator/chatbot/llm.py

@@ -39,7 +39,7 @@ class ChatbotLLMClient:
             if not api_key:
                 raise ValueError("GEMINI_API_KEY is not set in configuration for the chatbot agent.")
             
-            genai.configure(api_key=api_key)
+            self.client = genai.Client(api_key=api_key)
             self.model_name = settings.chatbot_agent.agent_model
             self._generate_func = self._generate_gemini
             
@@ -62,20 +62,21 @@ class ChatbotLLMClient:
 
     async def _generate_gemini(self, prompt: str, max_tokens: int) -> str:
         """Generates text using the Gemini model."""
-        generation_config = types.GenerationConfig(
+        generation_config = types.GenerateContentConfig(
             max_output_tokens=max_tokens,
         )
-        model = genai.GenerativeModel(self.model_name)
         try:
-            response = await model.generate_content_async(
+            # Run the synchronous SDK call in a thread to avoid blocking asyncio
+            response = await asyncio.to_thread(
+                self.client.models.generate_content,
+                model=self.model_name,
                 contents=prompt,
-                generation_config=generation_config
+                config=generation_config
             )
             return response.text
         except Exception as e:
             logger.error(f"Error in chatbot _generate_gemini: {e}", exc_info=True)
             return ""
-
     async def _generate_openai(self, prompt: str, max_tokens: int) -> str:
         """Generates text using the OpenAI model."""
         try:

{neuro_simulator-0.5.0.dist-info → neuro_simulator-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuro_simulator
-Version: 0.5.0
+Version: 0.5.1
 Summary: Neuro Simulator Server
 Author-email: Moha-Master <hongkongreporter@outlook.com>
 License-Expression: MIT

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

@@ -27,7 +27,7 @@ neuro_simulator/api/system.py,sha256=OJT6m6HIYATMCAKrgeBRhficaiUjIDl9f-WyUT-RoBw
 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/llm.py,sha256=SDFgPG4SU7CC--IYHyUHjUPK4no4vkzckIi47JZ6Vz4,4315
 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
@@ -55,8 +55,8 @@ neuro_simulator/utils/process.py,sha256=9OYWx8fzaJZqmFUcjQX37AnBhl7YWvrLxDWBa30v
 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.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-0.5.1.dist-info/METADATA,sha256=hHuiHZ9ZkV9Uu7tO71-lpH3y72088tQ_EN4hmla6jaA,7675
+neuro_simulator-0.5.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+neuro_simulator-0.5.1.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
+neuro_simulator-0.5.1.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
+neuro_simulator-0.5.1.dist-info/RECORD,,