Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
33 commits
Select commit Hold shift + click to select a range
3baecaa
Added monkey patch to pass in timeout for cohere
pk-zipstack Mar 10, 2026
f909a03
[FIX] Monkey-patch litellm cohere embed handler to forward timeout
pk-zipstack Mar 11, 2026
cada9d4
[FIX] Address review: version guard skips patch, stricter test assert…
pk-zipstack Mar 11, 2026
dcd0663
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Mar 11, 2026
079b91f
[FIX] Address SonarCloud warnings in cohere timeout patch
pk-zipstack Mar 11, 2026
b93ac45
Merge branch 'main' into fix/litellm-cohere-embed-timeout
pk-zipstack Mar 11, 2026
3f5e852
[FIX] Defer private litellm imports behind version guard
pk-zipstack Mar 12, 2026
f6181a6
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Mar 12, 2026
3a6cc0f
Apply suggestion from @hari-kuriakose
pk-zipstack Mar 12, 2026
644da84
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Mar 12, 2026
419d5e9
Update litellm patch version guard to 1.81.7
pk-zipstack Mar 12, 2026
89b8984
[FIX] Restore missing module imports and add production path test
pk-zipstack Mar 12, 2026
7b4c442
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Mar 12, 2026
97cc3e0
Added todo to remove this patch when the issue is fixed
pk-zipstack Mar 13, 2026
1f1fc3c
[FIX] Use DeprecationWarning for version guard skip message
pk-zipstack Mar 13, 2026
2b766f9
[FIX] Replace vacuous importlib.import_module test with sys.modules c…
pk-zipstack Mar 13, 2026
a9f662f
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Mar 13, 2026
25ed96c
Merge branch 'main' into fix/litellm-cohere-embed-timeout
pk-zipstack Mar 13, 2026
a97fd80
Merge branch 'main' into fix/litellm-cohere-embed-timeout
pk-zipstack Mar 13, 2026
94f1d65
Update unstract/sdk1/src/unstract/sdk1/patches/litellm_cohere_timeout.py
pk-zipstack Mar 13, 2026
2859264
[FIX] Mock validate_environment in sync timeout tests
pk-zipstack Mar 13, 2026
992dacd
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Mar 13, 2026
d86fac5
[FIX] Use pytest-asyncio for async test instead of asyncio.run()
pk-zipstack Mar 13, 2026
4e163dd
Commit uv.lock changes
pk-zipstack Mar 13, 2026
1c4b7ed
Merge branch 'main' into fix/litellm-cohere-embed-timeout
pk-zipstack Mar 18, 2026
1a33282
[FIX] Use logger.warning instead of DeprecationWarning for patch skip
pk-zipstack Mar 18, 2026
a71c9fe
[FIX] Remove async test cases and pytest-asyncio dependency
pk-zipstack Mar 18, 2026
136a0e0
[FIX] Restore lock files and keep pytest-asyncio in pyproject.toml
pk-zipstack Mar 19, 2026
3472494
[FIX] Regenerate sdk1 uv.lock to match pyproject.toml
pk-zipstack Mar 19, 2026
3ca198b
Updated litellm patch version since it has been updated in an upstrea…
pk-zipstack Mar 23, 2026
f5d0b3b
Merge branch 'main' into fix/litellm-cohere-embed-timeout
kirtimanmishrazipstack Mar 23, 2026
13a46f9
Merge branch 'main' into fix/litellm-cohere-embed-timeout
pk-zipstack Mar 23, 2026
eb1a25d
Merge branch 'fix/litellm-cohere-embed-timeout' of https://github.com…
pk-zipstack Mar 23, 2026
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions backend/uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions platform-service/uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

41 changes: 21 additions & 20 deletions prompt-service/uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions unstract/filesystem/uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions unstract/sdk1/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,7 @@ test = [
"pytest==8.3.3",
"pytest-asyncio>=0.23.0",
"pytest-mock==3.14.0",
"pytest-asyncio>=0.24.0",
"pytest-cov>=6.0.0",
"pytest-md-report>=0.6.2",
]
Expand Down
1 change: 1 addition & 0 deletions unstract/sdk1/src/unstract/sdk1/embedding.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
from typing import TYPE_CHECKING

import litellm
import unstract.sdk1.patches.litellm_cohere_timeout # noqa: F401
from llama_index.core.embeddings import BaseEmbedding
from pydantic import ValidationError
from unstract.sdk1.adapters.constants import Common
Expand Down
7 changes: 7 additions & 0 deletions unstract/sdk1/src/unstract/sdk1/patches/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
"""Monkey-patches for third-party library bugs.

Patches in this package are applied via side-effect imports.
Currently activated from unstract.sdk1.embedding — any code path
that reaches Bedrock Cohere embeddings without going through that
module will NOT have patches active.
"""
Loading
Loading