Spaces:
Sleeping
Sleeping
Add submission-safe defaults and project metadata
Browse files- inference.py +7 -4
- 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 =
|
| 314 |
-
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*"]
|