Skip to content

Commit db3d1b4

Browse files
committed
test: long timeout
1 parent 83aeda3 commit db3d1b4

File tree

3 files changed

+6
-5
lines changed

3 files changed

+6
-5
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[project]
22
name = "fenic"
3-
version = "0.5.0"
3+
version = "0.5.1"
44
authors = [{ name = "typedef", email = "[email protected]" }]
55
description = """
66
fenic is a Python DataFrame library for processing text data with APIs inspired by PySpark.

src/fenic/_inference/common_openai/openai_chat_completions_core.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,8 +94,9 @@ async def make_single_request(
9494
}
9595

9696
max_completion_tokens = self.get_max_output_token_request_limit(request, profile_configuration)
97-
if max_completion_tokens is not None:
98-
common_params["max_completion_tokens"] = max_completion_tokens
97+
# Don't set any limit for lineage builder
98+
#if max_completion_tokens is not None:
99+
# common_params["max_completion_tokens"] = max_completion_tokens
99100

100101
if request.temperature:
101102
common_params.update({"temperature": request.temperature})

src/fenic/_inference/model_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
RateLimitStrategy,
2727
TokenEstimate,
2828
)
29-
from fenic._inference.request_utils import DEFAULT_CLIENT_TIMEOUT
29+
from fenic._inference.request_utils import MAX_CLIENT_TIMEOUT
3030
from fenic._inference.token_counter import (
3131
TokenCounter,
3232
Tokenizable,
@@ -579,7 +579,7 @@ async def _process_single_request(self, queue_item: QueueItem[RequestT]):
579579
"""
580580
try:
581581
try:
582-
timeout = queue_item.request.request_timeout or DEFAULT_CLIENT_TIMEOUT
582+
timeout = MAX_CLIENT_TIMEOUT
583583
maybe_response = await asyncio.wait_for(
584584
self.make_single_request(queue_item.request),
585585
timeout=timeout,

0 commit comments

Comments
 (0)