Skip to content

Commit a155101

Browse files
committed
update tests
1 parent 2c43829 commit a155101

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

backend/src/shared/common_fn.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,8 +83,7 @@ def load_embedding_model(embedding_model_name: str):
8383
dimension = 1536
8484
logging.info(f"Embedding: Using bedrock titan Embeddings , Dimension:{dimension}")
8585
if embedding_model_name == "ecnu":
86-
ecnu_api_key=os.environ.get("ECNU_API_KEY")
87-
embeddings = OpenAIEmbeddings(base_url="https://chat.ecnu.edu.cn/open/api/v1",model="ecnu-embedding-small",api_key=ecnu_api_key,deployment="ecnu-embedding-small")
86+
embeddings = OpenAIEmbeddings(base_url="https://chat.ecnu.edu.cn/open/api/v1",model="ecnu-embedding-small",dimensions=1024,deployment="ecnu-embedding-small")
8887
dimension = 1024
8988
logging.info(f"Embedding: Using ECNU Embeddings , Dimension:{dimension}")
9089
else:

backend/test_commutiesqa.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ def test_populate_graph_schema_from_text(model):
202202
def run_tests():
203203
final_list = []
204204
error_list = []
205-
models = ['openai-gpt-3.5','openai-gpt-4o','openai-gpt-4o-mini','gemini-1.5-pro','azure_ai_gpt_35','azure_ai_gpt_4o','ollama_llama3','groq_llama3_70b','anthropic_claude_3_5_sonnet','fireworks_v3p1_405b','bedrock_claude_3_5_sonnet']
205+
models = ['ecnu_ecnu-plus','ecnu_ecnu-reasoner','ecnu_ecnu-reasoner-lite']
206206

207207
for model_name in models:
208208
try:

backend/test_integrationqa.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def run_tests():
178178
chatbot_list = []
179179
chatbot_error_list = []
180180
other_api_list = []
181-
models = ['openai_gpt_4','openai_gpt_4o','openai_gpt_4o_mini','gemini_1.5_pro','gemini_1.5_flash','gemini_2.0_flash','bedrock_nova_micro_v1','bedrock_nova_lite_v1','bedrock_nova_pro_v1','fireworks_qwen72b_instruct']
181+
models = ['ecnu_ecnu-plus','ecnu_ecnu-reasoner','ecnu_ecnu-reasoner-lite']
182182
chatbot_modes = [
183183
"vector",
184184
"graph+vector",

0 commit comments

Comments
 (0)