5ivatej commited on
Commit
e1ec6bc
·
1 Parent(s): 9f8b02c

Add submission-safe defaults and project metadata

Browse files
Files changed (2) hide show
  1. inference.py +7 -4
  2. pyproject.toml +23 -0
inference.py CHANGED
@@ -2,8 +2,8 @@
2
 
3
  MANDATORY env vars
4
  ------------------
5
- API_BASE_URL - LLM endpoint
6
- MODEL_NAME - Model identifier
7
  HF_TOKEN - Hugging Face / router token (preferred)
8
  ESC_ENV_URL - URL of the running ESC OpenEnv HTTP server (defaults to localhost)
9
 
@@ -66,6 +66,9 @@ SYSTEM_PROMPT = textwrap.dedent(
66
  """
67
  ).strip()
68
 
 
 
 
69
 
70
  def require_env(name: str) -> str:
71
  value = os.getenv(name)
@@ -310,8 +313,8 @@ async def run_task(
310
  # -------------------------- main ---------------------------------------------
311
 
312
  async def main() -> None:
313
- api_base_url = require_env("API_BASE_URL")
314
- model_name = require_env("MODEL_NAME")
315
  api_key = resolve_api_key()
316
  env_url = os.getenv("ESC_ENV_URL") or "http://127.0.0.1:7860"
317
 
 
2
 
3
  MANDATORY env vars
4
  ------------------
5
+ API_BASE_URL - LLM endpoint (defaults to https://api.openai.com/v1)
6
+ MODEL_NAME - Model identifier (defaults to gpt-4.1-mini)
7
  HF_TOKEN - Hugging Face / router token (preferred)
8
  ESC_ENV_URL - URL of the running ESC OpenEnv HTTP server (defaults to localhost)
9
 
 
66
  """
67
  ).strip()
68
 
69
+ DEFAULT_API_BASE_URL = "https://api.openai.com/v1"
70
+ DEFAULT_MODEL_NAME = "gpt-4.1-mini"
71
+
72
 
73
  def require_env(name: str) -> str:
74
  value = os.getenv(name)
 
313
  # -------------------------- main ---------------------------------------------
314
 
315
  async def main() -> None:
316
+ api_base_url = os.getenv("API_BASE_URL") or DEFAULT_API_BASE_URL
317
+ model_name = os.getenv("MODEL_NAME") or DEFAULT_MODEL_NAME
318
  api_key = resolve_api_key()
319
  env_url = os.getenv("ESC_ENV_URL") or "http://127.0.0.1:7860"
320
 
pyproject.toml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [build-system]
2
+ requires = ["setuptools>=68", "wheel"]
3
+ build-backend = "setuptools.build_meta"
4
+
5
+ [project]
6
+ name = "emotional-support-conversations"
7
+ version = "0.1.0"
8
+ description = "OpenEnv environment for evaluating agents on emotional support conversations."
9
+ readme = "README.md"
10
+ requires-python = ">=3.11"
11
+ dependencies = [
12
+ "fastapi==0.115.0",
13
+ "uvicorn[standard]==0.30.6",
14
+ "pydantic==2.9.2",
15
+ "httpx==0.27.2",
16
+ "openai==1.54.3",
17
+ ]
18
+
19
+ [tool.setuptools]
20
+ py-modules = ["server", "inference", "benchmark", "benchmark_agentic", "benchmark_agentic_llm", "benchmark_llm"]
21
+
22
+ [tool.setuptools.packages.find]
23
+ include = ["src*"]