mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-14 08:17:34 +00:00
fix: cleanup models, better retry strat
This commit is contained in:
1 parent
c3d605f07a
commit
32597df153
20 files changed
+849
-671
No files matched your search
@@ -497,24 +497,6 @@ class TestMarkdownParserEmptyHeader:
|
||||
assert len(tups) >= 1
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# application/llm/sagemaker.py (line 52)
|
||||
# ---------------------------------------------------------------------------
|
||||
@pytest.mark.unit
|
||||
class TestSagemakerLineIteratorStopIteration:
|
||||
def test_stop_iteration_with_newline_data(self):
|
||||
"""Cover line 52: chunk with newline yields a line."""
|
||||
from application.llm.sagemaker import LineIterator
|
||||
|
||||
# Chunk with newline so it yields
|
||||
chunks = [
|
||||
{"PayloadPart": {"Bytes": b'{"outputs": [" partial"]}\n'}},
|
||||
]
|
||||
it = LineIterator(iter(chunks))
|
||||
lines = list(it)
|
||||
assert len(lines) == 1
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# application/core/url_validation.py (lines 89-90)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
Reference in new issue
Block a user