Commit 77c29417 authored by Kantz's avatar Kantz
Browse files

ollama für LLMs und GWDG für outisde material

parent d040fafa
...@@ -23,7 +23,8 @@ def needs_more_context(history: str, context_sheet: str) -> dict: ...@@ -23,7 +23,8 @@ def needs_more_context(history: str, context_sheet: str) -> dict:
messages=[ messages=[
{"role": "system", "content": CLASSIFIER_PROMPT}, {"role": "system", "content": CLASSIFIER_PROMPT},
{"role": "user", "content": prompt}, {"role": "user", "content": prompt},
] ],
use_ollama=True,
) )
content = llm_client.get_message_content(result) content = llm_client.get_message_content(result)
try: try:
......
...@@ -17,7 +17,8 @@ def extract_math_request(user_text: str) -> dict | None: ...@@ -17,7 +17,8 @@ def extract_math_request(user_text: str) -> dict | None:
messages=[ messages=[
{"role": "system", "content": MATH_INTENT_PROMPT}, {"role": "system", "content": MATH_INTENT_PROMPT},
{"role": "user", "content": user_text}, {"role": "user", "content": user_text},
] ],
use_ollama=True,
) )
content = llm_client.get_message_content(result) content = llm_client.get_message_content(result)
try: try:
......
...@@ -29,8 +29,12 @@ def _chat_openai(messages: list[dict]) -> dict: ...@@ -29,8 +29,12 @@ def _chat_openai(messages: list[dict]) -> dict:
return {"raw": response, "message": message} return {"raw": response, "message": message}
def chat(messages: list[dict], tools: list[dict] | None = None) -> dict: def chat(
if not tools: messages: list[dict],
tools: list[dict] | None = None,
use_ollama: bool = False,
) -> dict:
if not tools and not use_ollama:
openai_result = _chat_openai(messages) openai_result = _chat_openai(messages)
if openai_result: if openai_result:
return openai_result return openai_result
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment