Revanth-ml commited on
Commit
bdf2d43
·
verified ·
1 Parent(s): 7b993b3

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -51,11 +51,11 @@ from agentops_gym.models import ToolCall
51
  IMAGE_NAME = os.getenv("IMAGE_NAME")
52
  API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
53
  MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
54
- HF_TOKEN = os.getenv("HF_TOKEN")
55
  ENV_BASE_URL = os.getenv("ENV_BASE_URL", "http://localhost:8000")
56
 
57
  # Backward compatibility for API_KEY
58
- API_KEY = HF_TOKEN or os.getenv("OPENAI_API_KEY") or os.getenv("API_KEY")
59
 
60
  BENCHMARK = "agentops-gym"
61
  MAX_STEPS = 10
 
51
  IMAGE_NAME = os.getenv("IMAGE_NAME")
52
  API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
53
  MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
54
+ API_KEY = os.getenv("HF_TOKEN")
55
  ENV_BASE_URL = os.getenv("ENV_BASE_URL", "http://localhost:8000")
56
 
57
  # Backward compatibility for API_KEY
58
+ # API_KEY = os.getenv("OPENAI_API_KEY") or os.getenv("API_KEY")
59
 
60
  BENCHMARK = "agentops-gym"
61
  MAX_STEPS = 10