mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-06 02:30:41 -08:00
Code Issues Projects Releases Packages Wiki Activity

fix: Update reasoning statuses in ThinkingProcessor tests

Browse source 
- Modified `TestThinkingProcessor` unit tests in `thinking.py` to update expected results for reasoning statuses:
  - Changed `"Finished"` status to `""` in `test_thinking_end`, `test_thinking_start_and_end`, and `test_chunk_with_text_after_think`.

feat: Add model aliases to HuggingFaceMedia provider

- Imported `model_aliases` from `.models` in `HuggingFaceMedia.py`.
- Added `model_aliases` attribute to `HuggingFaceMedia` provider class.

feat: Add timeout argument to CLI API parser

- Introduced `--timeout` argument in `get_api_parser()` in `cli.py` with default value of `600` seconds.
- Passed `timeout` argument to `run_api_args()` function.
```
This commit is contained in:
hlohaus 2025-06-19 00:48:57 +02:00
parent d824d77d65
commit f5bd41c598
3 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
etc/unittest/thinking.py
View file

@ -31,7 +31,7 @@ class TestThinkingProcessor(unittest.TestCase):
def test_thinking_end(self):
start_time = time.time()
chunk = "token</think> content after"
expected_result = [Reasoning("token"), Reasoning(status="Finished", is_thinking="</think>"), " content after"]
expected_result = [Reasoning("token"), Reasoning(status="", is_thinking="</think>"), " content after"]
actual_time, actual_result = ThinkingProcessor.process_thinking_chunk(chunk, start_time)
self.assertEqual(actual_time, 0)
self.assertEqualReasoning(actual_result[0], expected_result[0])
@ -41,7 +41,7 @@ class TestThinkingProcessor(unittest.TestCase):
def test_thinking_start_and_end(self):
start_time = time.time()
chunk = "<think>token</think> content after"
expected_result = [Reasoning(status="🤔 Is thinking...", is_thinking="<think>"), Reasoning("token"), Reasoning(status="Finished", is_thinking="</think>"), " content after"]
expected_result = [Reasoning(status="🤔 Is thinking...", is_thinking="<think>"), Reasoning("token"), Reasoning(status="", is_thinking="</think>"), " content after"]
actual_time, actual_result = ThinkingProcessor.process_thinking_chunk(chunk, start_time)
self.assertEqual(actual_time, 0)
self.assertEqualReasoning(actual_result[0], expected_result[0])
@ -60,7 +60,7 @@ class TestThinkingProcessor(unittest.TestCase):
def test_chunk_with_text_after_think(self):
chunk = "Start <think>Middle</think>End"
expected_time = 0
expected_result = ["Start ", Reasoning(status="🤔 Is thinking...", is_thinking="<think>"), Reasoning("Middle"), Reasoning(status="Finished", is_thinking="</think>"), "End"]
expected_result = ["Start ", Reasoning(status="🤔 Is thinking...", is_thinking="<think>"), Reasoning("Middle"), Reasoning(status="", is_thinking="</think>"), "End"]
actual_time, actual_result = ThinkingProcessor.process_thinking_chunk(chunk)
self.assertEqual(actual_time, expected_time)
for i in range(1, len(expected_result)):