callanwu commited on
Commit
73b165a
1 Parent(s): e874442
Files changed (1) hide show
  1. src/agents/utils.py +3 -3
src/agents/utils.py CHANGED
@@ -44,9 +44,9 @@ else:
44
  def get_embedding(sentence):
45
  if embed_model_name in ["text-embedding-ada-002"]:
46
  openai.api_key = os.environ["API_KEY"]
47
- if "PROXY" in os.environ:
48
- assert "http:" in os.environ["PROXY"] or "socks" in os.environ["PROXY"],"PROXY error,PROXY must be http or socks"
49
- openai.proxy = os.environ["PROXY"]
50
  if "API_BASE" in os.environ:
51
  openai.api_base = os.environ["API_BASE"]
52
  embedding_model = openai.Embedding
 
44
  def get_embedding(sentence):
45
  if embed_model_name in ["text-embedding-ada-002"]:
46
  openai.api_key = os.environ["API_KEY"]
47
+ # if "PROXY" in os.environ:
48
+ # assert "http:" in os.environ["PROXY"] or "socks" in os.environ["PROXY"],"PROXY error,PROXY must be http or socks"
49
+ # openai.proxy = os.environ["PROXY"]
50
  if "API_BASE" in os.environ:
51
  openai.api_base = os.environ["API_BASE"]
52
  embedding_model = openai.Embedding