minor update

Author: theonlyamos

cognitrix/agents/base.py

@@ -228,8 +228,11 @@ def initialize(self, session_id: Optional[str] = None):
                 logger.exception(e)
                 break
 
-    def start(self, session_id: Optional[str] = None):
-        self.initialize(session_id)
+    def start(self, session_id: Optional[str] = None, audio: bool = False):
+        if audio:
+            self.start_audio()
+        else:
+            self.initialize(session_id)
 
     def handle_transcription(self, sentence: str, transcriber: Transcriber):
         if sentence:

cognitrix/cli.py

@@ -124,10 +124,7 @@ def start(args: Namespace):
             assistant.format_system_prompt()
             asyncio.run(assistant.save())
 
-            if args.audio:
-                assistant.start_audio()
-            else:
-                assistant.start(args.session)
+            assistant.start(args.session, audio=args.audio)
     except Exception as e:
         logging.exception(e)
         parser.print_help()

cognitrix/config.py

@@ -3,7 +3,7 @@
 import aiofiles
 import json
 
-VERSION = '0.2.2'
+VERSION = '0.2.3'
 SPIRAL_WORKDIR = Path('~').expanduser() / '.cognitrix'
 
 AGENTS_FILE = SPIRAL_WORKDIR / 'agents.json'

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cognitrix"
-version = "0.2.2"
+version = "0.2.3"
 description = "Package for creating AI Agents using llms"
 authors = ["theonlyamos <[email protected]>"]
 license = "Apache2"