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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of auto-coder might be problematic. Click here for more details.

Files changed (138) hide show
  1. {auto_coder-0.1.213 → auto_coder-0.1.215}/PKG-INFO +2 -2
  2. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/auto_coder.egg-info/PKG-INFO +2 -2
  3. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/auto_coder.egg-info/requires.txt +1 -1
  4. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/chat_auto_coder.py +22 -47
  5. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/chat_auto_coder_lang.py +4 -4
  6. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/search.py +3 -6
  7. auto_coder-0.1.215/src/autocoder/version.py +1 -0
  8. auto_coder-0.1.213/src/autocoder/version.py +0 -1
  9. {auto_coder-0.1.213 → auto_coder-0.1.215}/LICENSE +0 -0
  10. {auto_coder-0.1.213 → auto_coder-0.1.215}/README.md +0 -0
  11. {auto_coder-0.1.213 → auto_coder-0.1.215}/setup.cfg +0 -0
  12. {auto_coder-0.1.213 → auto_coder-0.1.215}/setup.py +0 -0
  13. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/auto_coder.egg-info/SOURCES.txt +0 -0
  14. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/auto_coder.egg-info/dependency_links.txt +0 -0
  15. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/auto_coder.egg-info/entry_points.txt +0 -0
  16. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/auto_coder.egg-info/top_level.txt +0 -0
  17. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/__init__.py +0 -0
  18. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/__init__.py +0 -0
  19. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/auto_demand_organizer.py +0 -0
  20. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/auto_filegroup.py +0 -0
  21. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/auto_guess_query.py +0 -0
  22. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/auto_tool.py +0 -0
  23. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/coder.py +0 -0
  24. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/designer.py +0 -0
  25. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/planner.py +0 -0
  26. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/agent/project_reader.py +0 -0
  27. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/auto_coder.py +0 -0
  28. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/auto_coder_lang.py +0 -0
  29. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/auto_coder_rag.py +0 -0
  30. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/auto_coder_server.py +0 -0
  31. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/benchmark.py +0 -0
  32. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/chat/__init__.py +0 -0
  33. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/command_args.py +0 -0
  34. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/JupyterClient.py +0 -0
  35. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/ShellClient.py +0 -0
  36. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/__init__.py +0 -0
  37. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/anything2images.py +0 -0
  38. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/audio.py +0 -0
  39. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/buildin_tokenizer.py +0 -0
  40. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/chunk_validation.py +0 -0
  41. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/cleaner.py +0 -0
  42. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_execute.py +0 -0
  43. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_generate.py +0 -0
  44. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_generate_diff.py +0 -0
  45. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_generate_editblock.py +0 -0
  46. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_generate_strict_diff.py +0 -0
  47. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_merge.py +0 -0
  48. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_merge_diff.py +0 -0
  49. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_merge_editblock.py +0 -0
  50. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_auto_merge_strict_diff.py +0 -0
  51. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/code_modification_ranker.py +0 -0
  52. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/command_completer.py +0 -0
  53. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/command_generator.py +0 -0
  54. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/command_templates.py +0 -0
  55. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/const.py +0 -0
  56. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/git_utils.py +0 -0
  57. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/image_to_page.py +0 -0
  58. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/interpreter.py +0 -0
  59. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/llm_rerank.py +0 -0
  60. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/mcp_hub.py +0 -0
  61. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/mcp_server.py +0 -0
  62. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/mcp_tools.py +0 -0
  63. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/recall_validation.py +0 -0
  64. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/screenshots.py +0 -0
  65. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/search_replace.py +0 -0
  66. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/sys_prompt.py +0 -0
  67. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/text.py +0 -0
  68. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/common/types.py +0 -0
  69. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/data/tokenizer.json +0 -0
  70. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/db/__init__.py +0 -0
  71. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/db/store.py +0 -0
  72. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/dispacher/__init__.py +0 -0
  73. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/dispacher/actions/__init__.py +0 -0
  74. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/dispacher/actions/action.py +0 -0
  75. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/dispacher/actions/copilot.py +0 -0
  76. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/dispacher/actions/plugins/__init__.py +0 -0
  77. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/dispacher/actions/plugins/action_regex_project.py +0 -0
  78. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/dispacher/actions/plugins/action_translate.py +0 -0
  79. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/index/__init__.py +0 -0
  80. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/index/for_command.py +0 -0
  81. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/index/index.py +0 -0
  82. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/index/symbols_utils.py +0 -0
  83. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/lang.py +0 -0
  84. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/pyproject/__init__.py +0 -0
  85. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/__init__.py +0 -0
  86. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/api_server.py +0 -0
  87. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/cache/__init__.py +0 -0
  88. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/cache/base_cache.py +0 -0
  89. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/cache/byzer_storage_cache.py +0 -0
  90. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/cache/file_monitor_cache.py +0 -0
  91. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/cache/simple_cache.py +0 -0
  92. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/doc_filter.py +0 -0
  93. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/document_retriever.py +0 -0
  94. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/llm_wrapper.py +0 -0
  95. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/loaders/__init__.py +0 -0
  96. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/loaders/docx_loader.py +0 -0
  97. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/loaders/excel_loader.py +0 -0
  98. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/loaders/pdf_loader.py +0 -0
  99. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/loaders/ppt_loader.py +0 -0
  100. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/long_context_rag.py +0 -0
  101. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/rag_config.py +0 -0
  102. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/rag_entry.py +0 -0
  103. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/raw_rag.py +0 -0
  104. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/relevant_utils.py +0 -0
  105. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/simple_directory_reader.py +0 -0
  106. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/simple_rag.py +0 -0
  107. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/stream_event/__init__.py +0 -0
  108. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/stream_event/event_writer.py +0 -0
  109. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/stream_event/types.py +0 -0
  110. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/token_checker.py +0 -0
  111. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/token_counter.py +0 -0
  112. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/token_limiter.py +0 -0
  113. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/types.py +0 -0
  114. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/utils.py +0 -0
  115. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/rag/variable_holder.py +0 -0
  116. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/regexproject/__init__.py +0 -0
  117. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/suffixproject/__init__.py +0 -0
  118. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/tsproject/__init__.py +0 -0
  119. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/__init__.py +0 -0
  120. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/_markitdown.py +0 -0
  121. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/conversation_store.py +0 -0
  122. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/llm_client_interceptors.py +0 -0
  123. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/log_capture.py +0 -0
  124. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/multi_turn.py +0 -0
  125. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/operate_config_api.py +0 -0
  126. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/print_table.py +0 -0
  127. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/queue_communicate.py +0 -0
  128. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/request_event_queue.py +0 -0
  129. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/request_queue.py +0 -0
  130. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/rest.py +0 -0
  131. {auto_coder-0.1.213 → auto_coder-0.1.215}/src/autocoder/utils/tests.py +0 -0
  132. {auto_coder-0.1.213 → auto_coder-0.1.215}/tests/test_action_regex_project.py +0 -0
  133. {auto_coder-0.1.213 → auto_coder-0.1.215}/tests/test_chat_auto_coder.py +0 -0
  134. {auto_coder-0.1.213 → auto_coder-0.1.215}/tests/test_code_auto_merge_editblock.py +0 -0
  135. {auto_coder-0.1.213 → auto_coder-0.1.215}/tests/test_command_completer.py +0 -0
  136. {auto_coder-0.1.213 → auto_coder-0.1.215}/tests/test_planner.py +0 -0
  137. {auto_coder-0.1.213 → auto_coder-0.1.215}/tests/test_queue_communicate.py +0 -0
  138. {auto_coder-0.1.213 → auto_coder-0.1.215}/tests/test_symbols_utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: auto-coder
3
- Version: 0.1.213
3
+ Version: 0.1.215
4
4
  Summary: AutoCoder: AutoCoder
5
5
  Author: allwefantasy
6
6
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
26
26
  Requires-Dist: jupyter_client
27
27
  Requires-Dist: prompt-toolkit
28
28
  Requires-Dist: tokenizers
29
- Requires-Dist: byzerllm[saas]>=0.1.143
29
+ Requires-Dist: byzerllm[saas]>=0.1.144
30
30
  Requires-Dist: patch
31
31
  Requires-Dist: diff_match_patch
32
32
  Requires-Dist: GitPython
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: auto-coder
3
- Version: 0.1.213
3
+ Version: 0.1.215
4
4
  Summary: AutoCoder: AutoCoder
5
5
  Author: allwefantasy
6
6
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
26
26
  Requires-Dist: jupyter_client
27
27
  Requires-Dist: prompt-toolkit
28
28
  Requires-Dist: tokenizers
29
- Requires-Dist: byzerllm[saas]>=0.1.143
29
+ Requires-Dist: byzerllm[saas]>=0.1.144
30
30
  Requires-Dist: patch
31
31
  Requires-Dist: diff_match_patch
32
32
  Requires-Dist: GitPython
@@ -16,7 +16,7 @@ tabulate
16
16
  jupyter_client
17
17
  prompt-toolkit
18
18
  tokenizers
19
- byzerllm[saas]>=0.1.143
19
+ byzerllm[saas]>=0.1.144
20
20
  patch
21
21
  diff_match_patch
22
22
  GitPython
@@ -314,54 +314,29 @@ def initialize_system():
314
314
  except subprocess.CalledProcessError:
315
315
  print_status(get_message("model_error"), "error")
316
316
 
317
- # If deepseek_chat is not available, prompt user to choose a provider
318
- print_status(get_message("model_not_available"), "warning")
319
- choice = radiolist_dialog(
320
- title=get_message("provider_selection"),
321
- text=get_message("provider_selection"),
322
- values=[
323
- ("1", "硅基流动(https://siliconflow.cn)"),
324
- ("2", "Deepseek官方(https://www.deepseek.com/)"),
325
- ],
326
- ).run()
327
-
328
- if choice is None:
329
- print_status(get_message("no_provider"), "error")
330
- return
331
-
317
+ # If deepseek_chat is not available
318
+ print_status(get_message("model_not_available"), "warning")
332
319
  api_key = prompt(HTML(f"<b>{get_message('enter_api_key')} </b>"))
333
-
334
- if choice == "1":
335
- print_status(get_message("deploying_model").format("硅基流动"), "")
336
- deploy_cmd = [
337
- "easy-byzerllm",
338
- "deploy",
339
- "deepseek-ai/deepseek-v2-chat",
340
- "--token",
341
- api_key,
342
- "--alias",
343
- "deepseek_chat",
344
- ]
345
- else:
346
- print_status(get_message("deploying_model").format("Deepseek官方"), "")
347
- deploy_cmd = [
348
- "byzerllm",
349
- "deploy",
350
- "--pretrained_model_type",
351
- "saas/openai",
352
- "--cpus_per_worker",
353
- "0.001",
354
- "--gpus_per_worker",
355
- "0",
356
- "--worker_concurrency",
357
- "1000",
358
- "--num_workers",
359
- "1",
360
- "--infer_params",
361
- f"saas.base_url=https://api.deepseek.com/v1 saas.api_key={api_key} saas.model=deepseek-chat",
362
- "--model",
363
- "deepseek_chat",
364
- ]
320
+
321
+ print_status(get_message("deploying_model").format("Deepseek官方"), "")
322
+ deploy_cmd = [
323
+ "byzerllm",
324
+ "deploy",
325
+ "--pretrained_model_type",
326
+ "saas/openai",
327
+ "--cpus_per_worker",
328
+ "0.001",
329
+ "--gpus_per_worker",
330
+ "0",
331
+ "--worker_concurrency",
332
+ "1000",
333
+ "--num_workers",
334
+ "1",
335
+ "--infer_params",
336
+ f"saas.base_url=https://api.deepseek.com/v1 saas.api_key={api_key} saas.model=deepseek-chat",
337
+ "--model",
338
+ "deepseek_chat",
339
+ ]
365
340
 
366
341
  try:
367
342
  subprocess.run(deploy_cmd, check=True)
@@ -20,10 +20,10 @@ MESSAGES = {
20
20
  "model_available": "deepseek_chat model is available.",
21
21
  "model_timeout": "Command timed out. deepseek_chat model might not be available.",
22
22
  "model_error": "Error occurred while checking deepseek_chat model.",
23
- "model_not_available": "deepseek_chat model is not available. Please choose a provider:",
23
+ "model_not_available": "deepseek_chat model is not available.",
24
24
  "provider_selection": "Select a provider for deepseek_chat model:",
25
25
  "no_provider": "No provider selected. Exiting initialization.",
26
- "enter_api_key": "Please enter your API key: ",
26
+ "enter_api_key": "Please enter your API key(https://www.deepseek.com/): ",
27
27
  "deploying_model": "Deploying deepseek_chat model using {}...",
28
28
  "deploy_complete": "Deployment completed.",
29
29
  "deploy_fail": "Deployment failed. Please try again or deploy manually.",
@@ -86,10 +86,10 @@ MESSAGES = {
86
86
  "model_available": "deepseek_chat模型可用。",
87
87
  "model_timeout": "命令超时。deepseek_chat模型可能不可用。",
88
88
  "model_error": "检查deepseek_chat模型时出错。",
89
- "model_not_available": "deepseek_chat模型不可用。请选择一个提供商:",
89
+ "model_not_available": "deepseek_chat模型不可用。",
90
90
  "provider_selection": "为deepseek_chat模型选择一个提供商:",
91
91
  "no_provider": "未选择提供商。退出初始化。",
92
- "enter_api_key": "请输入您的API密钥:",
92
+ "enter_api_key": "请输入您的API密钥(https://www.deepseek.com/):",
93
93
  "deploying_model": "正在使用{}部署deepseek_chat模型...",
94
94
  "deploy_complete": "部署完成。",
95
95
  "deploy_fail": "部署失败。请重试或手动部署。",
@@ -6,7 +6,7 @@ from pydantic import BaseModel,Field
6
6
  import requests
7
7
  from enum import Enum
8
8
  import byzerllm
9
- from langchain_core.prompts import PromptTemplate
9
+ from byzerllm.utils import format_str_jinja2
10
10
  from autocoder.utils.rest import HttpDoc
11
11
 
12
12
  # Search engine related. You don't really need to change this.
@@ -51,10 +51,7 @@ def llm_rerank(llm:byzerllm.ByzerLLM,query:str,docs:List[str],top_k:int=1):
51
51
  "{context_str}\n"
52
52
  "Question: {query_str}\n"
53
53
  "Answer:\n"
54
- )
55
- DEFAULT_CHOICE_SELECT_PROMPT = PromptTemplate.from_template(
56
- DEFAULT_CHOICE_SELECT_PROMPT_TMPL
57
- )
54
+ )
58
55
 
59
56
  context_str = ""
60
57
  for i,metric in enumerate(docs):
@@ -64,7 +61,7 @@ def llm_rerank(llm:byzerllm.ByzerLLM,query:str,docs:List[str],top_k:int=1):
64
61
 
65
62
  r = llm.chat_oai(conversations=[{
66
63
  "role": "user",
67
- "content": DEFAULT_CHOICE_SELECT_PROMPT.format(context_str=context_str,query_str=query_str)
64
+ "content": format_str_jinja2(DEFAULT_CHOICE_SELECT_PROMPT_TMPL,context_str=context_str,query_str=query_str)
68
65
  }])
69
66
 
70
67
  r = llm.chat_oai(conversations=[{
@@ -0,0 +1 @@
1
+ __version__ = "0.1.215"
@@ -1 +0,0 @@
1
- __version__ = "0.1.213"
File without changes
File without changes
File without changes
File without changes