Skip to content

models: Add support for DeepSeek models #428

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.

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

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
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
10 changes: 10 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,7 @@ Support for various model providers:
```python
from strands import Agent
from strands.models import BedrockModel
from strands.models.deepseek import DeepSeekModel
from strands.models.ollama import OllamaModel
from strands.models.llamaapi import LlamaAPIModel

Expand All @@ -128,6 +129,14 @@ bedrock_model = BedrockModel(
agent = Agent(model=bedrock_model)
agent("Tell me about Agentic AI")

# DeepSeek
deepseek_model = DeepSeekModel(
api_key="your-deepseek-api-key",
model_id="deepseek-chat"
)
agent = Agent(model=deepseek_model)
agent("Tell me about Agentic AI")

# Ollama
ollama_model = OllamaModel(
host="http://localhost:11434",
Expand All @@ -147,6 +156,7 @@ response = agent("Tell me about Agentic AI")
Built-in providers:
- [Amazon Bedrock](https://strandsagents.com/latest/user-guide/concepts/model-providers/amazon-bedrock/)
- [Anthropic](https://strandsagents.com/latest/user-guide/concepts/model-providers/anthropic/)
- [DeepSeek](https://platform.deepseek.com/api-docs/)
- [LiteLLM](https://strandsagents.com/latest/user-guide/concepts/model-providers/litellm/)
- [LlamaAPI](https://strandsagents.com/latest/user-guide/concepts/model-providers/llamaapi/)
- [Ollama](https://strandsagents.com/latest/user-guide/concepts/model-providers/ollama/)
Expand Down
5 changes: 3 additions & 2 deletions src/strands/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,9 @@
This package includes an abstract base Model class along with concrete implementations for specific providers.
"""

from . import bedrock, model
from . import bedrock, deepseek, model
from .bedrock import BedrockModel
from .deepseek import DeepSeekModel
from .model import Model

__all__ = ["bedrock", "model", "BedrockModel", "Model"]
__all__ = ["bedrock", "deepseek", "model", "BedrockModel", "DeepSeekModel", "Model"]
Loading