diff --git a/README.md b/README.md index 81a6bf71..742ab4a2 100644 --- a/README.md +++ b/README.md @@ -100,8 +100,8 @@ runner = STORMWikiRunner(engine_args, lm_configs, rm) ``` Currently, our package support: -- `OpenAIModel`, `AzureOpenAIModel`, `ClaudeModel`, `VLLMClient`, `TGIClient`, `TogetherClient`, `OllamaClient`, `GoogleModel` as language model components -- `YouRM`, `BingSearch`, `VectorRM` as retrieval module components +- `OpenAIModel`, `AzureOpenAIModel`, `ClaudeModel`, `VLLMClient`, `TGIClient`, `TogetherClient`, `OllamaClient`, `GoogleModel`, `DeepSeekModel` as language model components +- `YouRM`, `BingSearch`, `VectorRM`, `SerperRM`, `BraveRM` as retrieval module components :star2: **PRs for integrating more language models into [knowledge_storm/lm.py](knowledge_storm/lm.py) and search engines/retrievers into [knowledge_storm/rm.py](knowledge_storm/rm.py) are highly appreciated!** diff --git a/knowledge_storm/rm.py b/knowledge_storm/rm.py index ce71d01b..4c438c47 100644 --- a/knowledge_storm/rm.py +++ b/knowledge_storm/rm.py @@ -3,7 +3,6 @@ from typing import Callable, Union, List import dspy -import pandas as pd import requests from langchain_huggingface import HuggingFaceEmbeddings diff --git a/setup.py b/setup.py index e4fa40c1..b7835ab7 100644 --- a/setup.py +++ b/setup.py @@ -16,7 +16,7 @@ setup( name="knowledge-storm", - version="0.2.3", + version="0.2.4", author="Yijia Shao, Yucheng Jiang", author_email="shaoyj@stanford.edu, yuchengj@stanford.edu", description="STORM: A language model-powered knowledge curation engine.",