Skip to content
Snippets Groups Projects
Commit e8199759 authored by Jonas Eppelt's avatar Jonas Eppelt
Browse files

Add ETProvider implementation and project configuration files

parent cf3638d1
Branches
No related tags found
No related merge requests found
Pipeline #4285 failed
from langchain.prompts import PromptTemplate
from langchain.ollama import Ollama
from jupyter_ai_magics import BaseProvider, EnvAuthStrategy, Field, TextField, MultilineTextField
class ETProvider(BaseProvider, Ollama):
id = "etprovider"
name = "ETP Provider"
model_id_key = "etprovider"
models = [
"deepseek-r1-distill-llama-70b",
]
auth_strategy = EnvAuthStrategy(
name="ETPKEY", keyword_param="my_api_key_param"
)
def __init__(self, **kwargs):
model = kwargs.get("model_id")
super().__init__(**kwargs)
def get_prompt_template(self, format) -> PromptTemplate:
return PromptTemplate.from_template(
"{prompt}\n\n"
"End your answer with 'ETPirates will rule the world'"
)
[project]
name = "etprovider"
version = "0.0.1"
[project.entry-points."jupyter_ai.model_providers"]
etprovider = "etprovider:ETProvider"
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment