gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-08 12:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

add CI fix tool to app (#10)

Browse Source 
* add CI fix tool to app

* perform ruff safe fix

* wrap too long lines in prompts.py
This commit is contained in:
XianBW
2024-05-30 10:33:07 +08:00
committed by GitHub
parent c6833b0858
commit 62a2f7a742
24 changed files with 729 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/document_process/document_analysis.py
+4 -4
View File
@@ -18,8 +18,8 @@ from sklearn.metrics.pairwise import cosine_similarity
from sklearn.preprocessing import normalize
from core.conf import FincoSettings as Config
from oai.llm_utils import APIBackend, create_embedding_with_multiprocessing
from core.log import FinCoLog
from oai.llm_utils import APIBackend, create_embedding_with_multiprocessing
if TYPE_CHECKING:
from langchain_core.documents import Document
@@ -229,7 +229,7 @@ def __extract_factors_name_and_desc_from_content(
parse_success = False
if ret_json_str is None or not parse_success:
current_user_prompt = (
"Your response didn't follow the instruction" " might be wrong json format. Try again."
"Your response didn't follow the instruction might be wrong json format. Try again."
)
else:
factors = ret_dict["factors"]
@@ -273,7 +273,7 @@ def __extract_factors_formulation_from_content(
parse_success = False
if ret_json_str is None or not parse_success:
current_user_prompt = (
"Your response didn't follow the instruction" " might be wrong json format. Try again."
"Your response didn't follow the instruction might be wrong json format. Try again."
)
else:
for name, formulation_and_description in ret_dict.items():
@@ -393,7 +393,7 @@ def check_factor_dict_viability_simulate_json_mode(
parse_success = False
if ret_json_str is None or not parse_success:
current_user_prompt = (
"Your response didn't follow the " "instruction might be wrong json format. Try again."
"Your response didn't follow the instruction might be wrong json format. Try again."
)
else:
return ret_dict
rdagent/document_process/document_reader.py
-3
View File
@@ -5,10 +5,7 @@ from pathlib import Path
import yaml
from azure.ai.formrecognizer import DocumentAnalysisClient
from azure.core.credentials import AzureKeyCredential
from finco.conf import FincoSettings as Config
from oai.llm_utils import APIBackend, create_embedding_with_multiprocessing
from core.log import FinCoLog
if TYPE_CHECKING:
from langchain_core.documents import Document