auto-coder 0.1.181__py3-none-any.whl → 0.1.183__py3-none-any.whl

{auto_coder-0.1.181.dist-info → auto_coder-0.1.183.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.181
+Version: 0.1.183
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
 Requires-Dist: jupyter-client
 Requires-Dist: prompt-toolkit
 Requires-Dist: tokenizers
-Requires-Dist: byzerllm[saas] >=0.1.134
+Requires-Dist: byzerllm[saas] >=0.1.136
 Requires-Dist: patch
 Requires-Dist: diff-match-patch
 Requires-Dist: GitPython

{auto_coder-0.1.181.dist-info → auto_coder-0.1.183.dist-info}/RECORD

@@ -1,13 +1,13 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/auto_coder.py,sha256=HmgKa_ZApFlCsqo6BvuVeCPuncBT_Dh29ayZxxGR6lo,32216
 autocoder/auto_coder_lang.py,sha256=4qIS1tbEI8mpbtt6ThppTwKOM6MLuJTWJdgs5jIDGE0,2301
-autocoder/auto_coder_rag.py,sha256=V82EyeslAO2Z8qkMrwkyC11f1Cz6Ccjo9c867f0J_x8,11455
+autocoder/auto_coder_rag.py,sha256=XGqpKmO7AVN4buAX-wuk1PgCGyvIiJwOf_kifhA3WDk,11765
 autocoder/auto_coder_server.py,sha256=qRY88mkBnqSGFDcwYE5gwpe2WPhIw1nEH6LdbjCQhQk,20306
-autocoder/chat_auto_coder.py,sha256=Yh9QRYsNe8_aQS9_R_K25WrNzLFS57sMDQTK6QKWR1U,81579
+autocoder/chat_auto_coder.py,sha256=_FzDNees7yp5bWpbOSTkEUYCp4iz3Ghjs8pJogrFPy8,81707
 autocoder/chat_auto_coder_lang.py,sha256=QYtu5gWEQmWKVovR_qUZ8plySZarNFX_Onk-1vN9IiA,8524
 autocoder/command_args.py,sha256=ftWw6HnFUZPiQPt1oV-SfpHQe69XN3knaFy1lpROBcU,26854
 autocoder/lang.py,sha256=e-07rYTgimpxS8sm-AxKSmH4kKQX4N05YFHJBg9trVs,12598
-autocoder/version.py,sha256=ZdxQFgfwMgcLstS2tHqv93Tm2T7lXzvvmJLNB0GKS_c,23
+autocoder/version.py,sha256=262HoV_V3P5Q9BFxC-7XwuMUEj7KKuFEc7fj0QupKz0,23
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_tool.py,sha256=DBzip-P_T6ZtT2eHexPcusmKYD0h7ufzp7TLwXAY10E,11554
 autocoder/agent/coder.py,sha256=dnITYHqkcOip8zV4lywbkYNH9w7Q3qyYaUArJ4WPrTs,866
@@ -17,7 +17,7 @@ autocoder/agent/project_reader.py,sha256=-MWRqsr7O4mvU0PIpAhOUBb29htZAvA37pa_GeE
 autocoder/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/common/JupyterClient.py,sha256=O-wi6pXeAEYhAY24kDa0BINrLYvKS6rKyWe98pDClS0,2816
 autocoder/common/ShellClient.py,sha256=fM1q8t_XMSbLBl2zkCNC2J9xuyKN3eXzGm6hHhqL2WY,2286
-autocoder/common/__init__.py,sha256=wKrFLZk9BMl755nL1gvPjXU-3uWKEnYBP8xsObIjM4g,10156
+autocoder/common/__init__.py,sha256=7vL4gbnZ_c3ktAZfEgdh5fDkbDjGfrD1ybylQjgLBGs,10262
 autocoder/common/anything2images.py,sha256=0ILBbWzY02M-CiWB-vzuomb_J1hVdxRcenAfIrAXq9M,25283
 autocoder/common/audio.py,sha256=Kn9nWKQddWnUrAz0a_ZUgjcu4VUU_IcZBigT7n3N3qc,7439
 autocoder/common/cleaner.py,sha256=NU72i8C6o9m0vXExab7nao5bstBUsfJFcj11cXa9l4U,1089
@@ -62,7 +62,7 @@ autocoder/rag/api_server.py,sha256=zokIlDJlk7ucRorSLQm80uICO1mecfmn4J2zVqEBskE,6
 autocoder/rag/doc_filter.py,sha256=Ha0Yae_G_hF72YzvrO7NoDZcG18K4hRcqGAEqfrIwAs,9330
 autocoder/rag/document_retriever.py,sha256=i_ilMTQiMg8gLK8YibJuRDc_7CVxER6bNR6-ZlcIgVI,23420
 autocoder/rag/llm_wrapper.py,sha256=xRbTBpLUH43Ah5jplL8WWWU-kjKfNgEJoUntLGBq5F4,2484
-autocoder/rag/long_context_rag.py,sha256=V8Z1LNekm6s1XO950yab1QEj9C4vLJv3qpGqhsRqmBo,20356
+autocoder/rag/long_context_rag.py,sha256=QA9RJ-NDBVl4W-gTs4ib_ZNs98EHSMAYHydt5QCsTq4,20509
 autocoder/rag/rag_config.py,sha256=8LwFcTd8OJWWwi1_WY4IzjqgtT6RyE2j4PjxS5cCTDE,802
 autocoder/rag/rag_entry.py,sha256=V1RJ8RGqM30DNPmzymv64rZjNRGWn6kfc8sRy_LECg0,2451
 autocoder/rag/raw_rag.py,sha256=yS2Ur6kG0IRjhCj2_VonwxjY_xls_E62jO5Gz5j2nqE,2952
@@ -95,9 +95,9 @@ autocoder/utils/request_event_queue.py,sha256=r3lo5qGsB1dIjzVQ05dnr0z_9Z3zOkBdP1
 autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-yPE,3889
 autocoder/utils/rest.py,sha256=3tXA8KZG6jKz_tddHNLGx77Icee88WcUeesfNsgPno4,8790
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
-auto_coder-0.1.181.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.181.dist-info/METADATA,sha256=6o-R-qsou7m4vPmbbw_bnEkSvPpUYAkYq0vE0SvWvTk,2352
-auto_coder-0.1.181.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.181.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.181.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.181.dist-info/RECORD,,
+auto_coder-0.1.183.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.183.dist-info/METADATA,sha256=kCcF3MOf3teH85V8JhNGYavQQczgFZtALK5s82BCBec,2352
+auto_coder-0.1.183.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.183.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.183.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.183.dist-info/RECORD,,

autocoder/auto_coder_rag.py

@@ -231,6 +231,17 @@ def main(input_args: Optional[List[str]] = None):
         help="Disable reordering of document segments after retrieval",
     )
 
+    serve_parser.add_argument(
+        "--disable_inference_enhance",
+        action="store_true",
+        help="Disable enhanced inference mode",
+    )
+    serve_parser.add_argument(
+        "--inference_deep_thought",
+        action="store_true",
+        help="Enable deep thought in inference mode",
+    )
+
     # Tools command
     tools_parser = subparsers.add_parser("tools", help="Various tools")
     tools_subparsers = tools_parser.add_subparsers(dest="tool", help="Available tools")

autocoder/chat_auto_coder.py

@@ -718,10 +718,11 @@ class CommandCompleter(Completer):
 
                 parser.coding()
                 current_word = parser.current_word()
-
-                for command in parser.get_sub_commands():
-                    if command.startswith(current_word):
-                        yield Completion(command, start_position=-len(current_word))
+                
+                if len(new_text.strip()) == 0 or new_text.strip()=="/":
+                    for command in parser.get_sub_commands():
+                        if command.startswith(current_word):
+                            yield Completion(command, start_position=-len(current_word))
 
                 all_tags = parser.tags
 
@@ -796,6 +797,8 @@ class CommandCompleter(Completer):
                                 start_position=-len(name),
                                 display=f"{symbol.symbol_name} ({display_name}/{symbol.symbol_type})",
                             )
+                
+                tags = [tag for tag in parser.tags]
 
                 if current_word.startswith("<"):
                     name = current_word[1:]
@@ -807,9 +810,7 @@ class CommandCompleter(Completer):
                                 )
                         elif tag.startswith(name):
                             yield Completion(tag, start_position=-len(current_word))
-
-                tags = [tag for tag in parser.tags]
-
+            
                 if tags and tags[-1].start_tag == "<img>" and tags[-1].end_tag == "":
                     raw_file_name = tags[0].content
                     file_name = raw_file_name.strip()

autocoder/common/__init__.py

@@ -312,5 +312,8 @@ class AutoCoderArgs(pydantic.BaseModel):
     segment_ratio: Optional[float] = 0.2
     buff_ratio: Optional[float] = 0.1
 
+    disable_inference_enhance: Optional[bool] = False
+    inference_deep_thought: Optional[bool] = False
+
     class Config:
         protected_namespaces = ()

autocoder/rag/long_context_rag.py

@@ -492,7 +492,10 @@ class LongContextRAG:
 
             if LLMComputeEngine is not None:
                 llm_compute_engine = LLMComputeEngine(
-                    llm=self.llm, inference_enhance=True, debug=False
+                    llm=self.llm,
+                    inference_enhance=not self.args.disable_inference_enhance,
+                    inference_deep_thought=self.args.inference_deep_thought,
+                    debug=False,
                 )
                 new_conversations = llm_compute_engine.process_conversation(
                     conversations, query, [doc.source_code for doc in relevant_docs]

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.181"
+__version__ = "0.1.183"