Add support for DeepSeek language models in STORM Wiki pipeline #84
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
This pull request adds support for using DeepSeek language models in the STORM Wiki pipeline, providing an alternative to the existing OpenAI models. The integration allows users to easily switch between OpenAI and DeepSeek models, enhancing the flexibility and capabilities of the STORM Wiki system.
Key Changes
DeepSeekModel
class insrc/lm.py
, which is compatible with the existingdspy.OpenAI
interface.examples/run_storm_wiki_deepseek.py
to demonstrate how to use DeepSeek models with the STORM Wiki pipeline.STORMWikiRunner
class to be model-agnostic, ensuring compatibility with both OpenAI and DeepSeek models.How to Use
Users can now run the STORM Wiki pipeline with DeepSeek models by using the new
run_storm_wiki_deepseek.py
script. The script allows configuration of various DeepSeek-specific parameters such as model choice, temperature, and top_p sampling. The API key should be set withDEEPSEEK_API_KEY
. The DeepSeek base can be set withDEEPSEEK_API_BASE
, if needed.Testing
Future Considerations