Pengfa Li commited on
Commit
43703cd
·
verified ·
1 Parent(s): 674554d

Update LLM.py

Browse files
Files changed (1) hide show
  1. LLM.py +4 -8
LLM.py CHANGED
@@ -4,20 +4,16 @@ import pandas as pd
4
  import google.generativeai as genai
5
  import requests
6
 
7
- # 输入模型系列和内容
8
  def zero_shot(model_series, model_name, content):
9
- # 根据模型系列获取对应的环境变量
10
  if model_series in ['gpt', 'gemini', 'claude']:
11
  api_key = os.environ.get("OPENAI_API_KEY")
12
  elif model_series in ['deepSeek', 'qwen', 'llama']:
13
  api_key = os.environ.get("DEEPSEEK_API_KEY")
14
  else:
15
- return "不支持的模型系列"
16
-
17
- if not api_key:
18
- return f"未找到 {model_series} 对应的API密钥环境变量"
19
 
20
- # 根据不同模型系列调用不同API
21
  if model_series == 'gpt':
22
  client = OpenAI(
23
  api_key=api_key,
@@ -84,4 +80,4 @@ def zero_shot(model_series, model_name, content):
84
  return content
85
 
86
  else:
87
- return "不支持的模型系列"
 
4
  import google.generativeai as genai
5
  import requests
6
 
 
7
  def zero_shot(model_series, model_name, content):
8
+
9
  if model_series in ['gpt', 'gemini', 'claude']:
10
  api_key = os.environ.get("OPENAI_API_KEY")
11
  elif model_series in ['deepSeek', 'qwen', 'llama']:
12
  api_key = os.environ.get("DEEPSEEK_API_KEY")
13
  else:
14
+ return "not support"
 
 
 
15
 
16
+
17
  if model_series == 'gpt':
18
  client = OpenAI(
19
  api_key=api_key,
 
80
  return content
81
 
82
  else:
83
+ return "not support"