Skip to content

Commit bb743b7

Browse files
committed
fixes
1 parent c699532 commit bb743b7

File tree

2 files changed

+1
-22
lines changed

2 files changed

+1
-22
lines changed

.github/workflows/streaming_compliance.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ jobs:
7676
env:
7777
FIREWORKS_API_KEY: ${{ secrets.FIREWORKS_API_KEY }}
7878
FIREWORKS_ACCOUNT_ID: ${{ vars.FIREWORKS_ACCOUNT_ID }}
79+
EP_DISABLE_DATASET_LOGGER: "1"
7980
run: |
8081
. .venv/bin/activate
8182
mkdir -p artifacts

eval_protocol/benchmarks/test_glm_streaming_compliance.py

Lines changed: 0 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -143,17 +143,6 @@ def _safe_json_loads(payload: str) -> Any | None:
143143
def test_glm_streaming_structured_output(row: EvaluationRow) -> EvaluationRow:
144144
"""Ensure structured output arrives in assistant content when streaming."""
145145

146-
import os
147-
148-
print(
149-
"DEBUG completion params",
150-
os.environ.get("EP_COMPLETION_PARAMS_JSON"),
151-
"key len",
152-
len(os.environ.get("FIREWORKS_API_KEY", "")),
153-
"acct",
154-
os.environ.get("FIREWORKS_ACCOUNT_ID"),
155-
)
156-
157146
assistant_msg = row.last_assistant_message()
158147
if assistant_msg is None:
159148
row.evaluation_result = EvaluateResult(
@@ -257,17 +246,6 @@ def test_glm_streaming_structured_output(row: EvaluationRow) -> EvaluationRow:
257246
def test_glm_streaming_tool_call(row: EvaluationRow) -> EvaluationRow:
258247
"""Ensure streaming tool calls settle with finish_reason=tool_calls and a single call."""
259248

260-
import os
261-
262-
print(
263-
"DEBUG completion params",
264-
os.environ.get("EP_COMPLETION_PARAMS_JSON"),
265-
"key len",
266-
len(os.environ.get("FIREWORKS_API_KEY", "")),
267-
"acct",
268-
os.environ.get("FIREWORKS_ACCOUNT_ID"),
269-
)
270-
271249
assistant_msg = row.last_assistant_message()
272250
if assistant_msg is None:
273251
row.evaluation_result = EvaluateResult(

0 commit comments

Comments
 (0)