Overview
Last updated
Was this helpful?
Last updated
Was this helpful?
Was this helpful?
from djangosdk.providers.base import AbstractProvider
from djangosdk.agents.request import AgentRequest
from djangosdk.agents.response import AgentResponse, StreamChunk
from typing import Iterator, AsyncIterator
class MyCustomProvider(AbstractProvider):
def complete(self, request: AgentRequest) -> AgentResponse:
...
async def acomplete(self, request: AgentRequest) -> AgentResponse:
...
def stream(self, request: AgentRequest) -> Iterator[StreamChunk]:
...
async def astream(self, request: AgentRequest) -> AsyncIterator[StreamChunk]:
...from djangosdk.providers.registry import registry
registry.register("my_provider", MyCustomProvider())