auto-coder 0.1.398__py3-none-any.whl → 0.1.400__py3-none-any.whl

autocoder/agent/entry_command_agent/project_reader.py

@@ -0,0 +1,58 @@
+
+
+import os
+from rich.console import Console
+from rich.panel import Panel
+from rich.markdown import Markdown
+from rich.live import Live
+
+from autocoder.privacy.model_filter import ModelPathFilter
+from autocoder.common.printer import Printer
+from autocoder.utils.llms import get_llm_names
+
+
+class ProjectReaderAgent:
+    def __init__(self, args, llm, raw_args):
+        self.args = args
+        self.llm = llm
+        self.raw_args = raw_args
+        self.console = Console()
+
+    def run(self):
+        """执行 project_reader 命令的主要逻辑"""
+        target_llm = self.llm.get_sub_client("planner_model")
+        if not target_llm:
+            target_llm = self.llm
+        
+        model_filter = ModelPathFilter.from_model_object(target_llm, self.args)
+        if model_filter.has_rules():
+            printer = Printer()
+            msg = printer.get_message_from_key_with_format(
+                "model_has_access_restrictions",                                            
+                model_name=",".join(get_llm_names(target_llm))
+            )
+            raise ValueError(msg)
+
+        from autocoder.agent.project_reader import ProjectReader
+
+        project_reader = ProjectReader(self.args, self.llm)
+        v = project_reader.run(self.args.query)            
+        
+        markdown_content = v
+
+        with Live(
+            Panel("", title="Response", border_style="green", expand=False),
+            refresh_per_second=4,
+            auto_refresh=True,
+            vertical_overflow="visible",
+            console=Console(force_terminal=True, color_system="auto", height=None)
+        ) as live:
+            live.update(
+                Panel(
+                    Markdown(markdown_content),
+                    title="Response",
+                    border_style="green",
+                    expand=False,
+                )
+            )
+

autocoder/agent/entry_command_agent/voice2text.py

@@ -0,0 +1,71 @@
+
+
+
+import os
+import tempfile
+from rich.console import Console
+from rich.panel import Panel
+
+from autocoder.utils.request_queue import (
+    request_queue,
+    RequestValue,
+    DefaultValue,
+    RequestOption,
+)
+
+
+class Voice2TextAgent:
+    def __init__(self, args, llm, raw_args):
+        self.args = args
+        self.llm = llm
+        self.raw_args = raw_args
+        self.console = Console()
+
+    def run(self):
+        """执行 voice2text 命令的主要逻辑"""
+        from autocoder.common.audio import TranscribeAudio
+
+        transcribe_audio = TranscribeAudio()
+        temp_wav_file = os.path.join(
+            tempfile.gettempdir(), "voice_input.wav")
+
+        transcribe_audio.record_audio(temp_wav_file)
+        self.console.print(
+            Panel(
+                "Recording finished. Transcribing...",
+                title="Voice",
+                border_style="green",
+            )
+        )
+
+        if self.llm and self.llm.get_sub_client("voice2text_model"):
+            voice2text_llm = self.llm.get_sub_client("voice2text_model")
+        else:
+            voice2text_llm = self.llm
+        
+        transcription = transcribe_audio.transcribe_audio(
+            temp_wav_file, voice2text_llm
+        )
+
+        self.console.print(
+            Panel(
+                f"Transcription: <_transcription_>{transcription}</_transcription_>",
+                title="Result",
+                border_style="magenta",
+            )
+        )
+
+        with open(os.path.join(".auto-coder", "exchange.txt"), "w", encoding="utf-8") as f:
+            f.write(transcription)
+
+        request_queue.add_request(
+            self.args.request_id,
+            RequestValue(
+                value=DefaultValue(value=transcription),
+                status=RequestOption.COMPLETED,
+            ),
+        )
+
+        os.remove(temp_wav_file)
+
+