Skip to content

Fix AzureOpenAIModel to match the latest Azure API. #309

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Jan 23, 2025
Merged

Fix AzureOpenAIModel to match the latest Azure API. #309

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
6f11450
Fix `AzureOpenAIModel` to match the latest Azure API.
shaoyijia Jan 15, 2025
0764f54
encapsulate encoding function and Encoder class
Yucheng-Jiang Jan 22, 2025
46b472b
integrate Litellm into lm interface
Yucheng-Jiang Jan 22, 2025
8820be3
Co-STORM code revision
Yucheng-Jiang Jan 22, 2025
a7b0be0
STORM code typo fix
Yucheng-Jiang Jan 22, 2025
e213650
update StanfordOvalArxivRM for internal usage
Yucheng-Jiang Jan 22, 2025
faff1f7
bump up version; add encoder class document; update readme
Yucheng-Jiang Jan 22, 2025
5fd4bc9
fix code style with black
Yucheng-Jiang Jan 22, 2025
068c9b9
fix code style
Yucheng-Jiang Jan 22, 2025
3362dae
Add litellm integration in `knowledge-storm` v1.1.0.
shaoyijia Jan 22, 2025
73bdddd
Patch `LitellmModel`.
shaoyijia Jan 22, 2025
ea39855
Specify `litellm` version in `requirements.txt`.
shaoyijia Jan 22, 2025
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
7 changes: 5 additions & 2 deletions knowledge_storm/lm.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@
litellm.drop_params = True
litellm.telemetry = False

from litellm.caching.caching import Cache
from litellm.caching import Cache

disk_cache_dir = os.path.join(Path.home(), ".storm_local_cache")
litellm.cache = Cache(disk_cache_dir=disk_cache_dir, type="disk")
Expand Down Expand Up @@ -190,7 +190,10 @@ def _inspect_history(lm, n: int = 1):


class LitellmModel(LM):
"""A wrapper class for dspy.OpenAI."""
"""A wrapper class for LiteLLM.

Check out https://docs.litellm.ai/docs/providers for usage details.
"""

def __init__(
self,
Expand Down
Loading