auto-coder 0.1.215__tar.gz → 0.1.218__tar.gz

{auto_coder-0.1.215 → auto_coder-0.1.218}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.215
+Version: 0.1.218
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -38,6 +38,7 @@ Requires-Dist: azure-cognitiveservices-speech
 Requires-Dist: python-docx
 Requires-Dist: docx2txt
 Requires-Dist: pdf2image
+Requires-Dist: Spire.Doc
 Requires-Dist: docx2pdf
 Requires-Dist: pyperclip
 Requires-Dist: colorama

{auto_coder-0.1.215 → auto_coder-0.1.218}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.215
+Version: 0.1.218
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -38,6 +38,7 @@ Requires-Dist: azure-cognitiveservices-speech
 Requires-Dist: python-docx
 Requires-Dist: docx2txt
 Requires-Dist: pdf2image
+Requires-Dist: Spire.Doc
 Requires-Dist: docx2pdf
 Requires-Dist: pyperclip
 Requires-Dist: colorama

{auto_coder-0.1.215 → auto_coder-0.1.218}/src/auto_coder.egg-info/SOURCES.txt

@@ -32,6 +32,7 @@ src/autocoder/common/JupyterClient.py
 src/autocoder/common/ShellClient.py
 src/autocoder/common/__init__.py
 src/autocoder/common/anything2images.py
+src/autocoder/common/anything2img.py
 src/autocoder/common/audio.py
 src/autocoder/common/buildin_tokenizer.py
 src/autocoder/common/chunk_validation.py
@@ -64,6 +65,8 @@ src/autocoder/common/search_replace.py
 src/autocoder/common/sys_prompt.py
 src/autocoder/common/text.py
 src/autocoder/common/types.py
+src/autocoder/common/mcp_servers/__init__.py
+src/autocoder/common/mcp_servers/mcp_server_perplexity.py
 src/autocoder/data/tokenizer.json
 src/autocoder/db/__init__.py
 src/autocoder/db/store.py

{auto_coder-0.1.215 → auto_coder-0.1.218}/src/auto_coder.egg-info/requires.txt

@@ -28,6 +28,7 @@ azure-cognitiveservices-speech
 python-docx
 docx2txt
 pdf2image
+Spire.Doc
 docx2pdf
 pyperclip
 colorama

{auto_coder-0.1.215 → auto_coder-0.1.218}/src/autocoder/auto_coder.py

@@ -786,7 +786,7 @@ def main(input_args: Optional[List[str]] = None):
                     pre_conversations.append(
                         {
                             "role": "user",
-                            "content": f"下面是一些文档和源码，如果用户的问题和他们相关，请参考他们：\n{file_content}",
+                            "content": f"下面是一些文档和源码，如果用户的问题和他们相关，请参考他们：\n <files>\n{file_content}</files>",
                         },
                     )
                     pre_conversations.append(
@@ -813,7 +813,7 @@ def main(input_args: Optional[List[str]] = None):
                 pre_conversations.append(
                     {
                         "role": "user",
-                        "content": f"下面是一些文档和源码，如果用户的问题和他们相关，请参考他们：\n{s}",
+                        "content": f"下面是一些文档和源码，如果用户的问题和他们相关，请参考他们：\n <files>{s}</files>",
                     }
                 )
                 pre_conversations.append(
@@ -831,15 +831,19 @@ def main(input_args: Optional[List[str]] = None):
                     source_codes, pre_conversations, last_conversation
                 ):
                     """
+                    <files>
                     {% if source_codes %}
                     {{ source_codes }}
                     {% endif %}
+                    </files>
 
                     {% if pre_conversations %}
                     下面是我们之间的历史对话，假设我是A，你是B。
+                    <conversations>
                     {% for conv in pre_conversations %}
                     {{ "A" if conv.role == "user" else "B" }}: {{ conv.content }}
                     {% endfor %}
+                    </conversations>
                     {% endif %}
 
 

{auto_coder-0.1.215 → auto_coder-0.1.218}/src/autocoder/auto_coder_rag.py

@@ -19,10 +19,12 @@ from rich.table import Table
 import os
 from loguru import logger
 import asyncio
+from datetime import datetime
 
 from autocoder.rag.document_retriever import process_file_local
 import pkg_resources
 from autocoder.rag.token_counter import TokenCounter
+from autocoder.rag.types import RAGServiceInfo
 
 if platform.system() == "Windows":
     from colorama import init
@@ -501,6 +503,20 @@ def main(input_args: Optional[List[str]] = None):
             rag = RAGFactory.get_rag(llm=llm, args=auto_coder_args, path="")
 
         llm_wrapper = LLWrapper(llm=llm, rag=rag)
+        # Save service info    
+        service_info = RAGServiceInfo(
+            host=server_args.host or "127.0.0.1",
+            port=server_args.port,
+            model=args.model,
+            _pid=os.getpid(),
+            _timestamp=datetime.now().strftime("%Y-%m-%d %H:%M:%S"),  
+            args={k: v for k, v in vars(args).items() if not k.startswith("_")}
+        )
+        try:
+            service_info.save()
+        except Exception as e:
+            logger.warning(f"Failed to save service info: {str(e)}")
+        
         serve(llm=llm_wrapper, args=server_args)
     elif args.command == "build_hybrid_index":
         if not args.quick: