Skip to content

Commit ca49bde

Browse files
committed
fixing authentication in chatting endpoint
1 parent 66de6e6 commit ca49bde

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

nedrexapi/llm.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,14 @@
1414
_LLM_chat_base=_config[f"chat.server_base"]
1515
_LLM_chat_api_key=_config[f"chat.api_key"]
1616

17-
headers = {"Authorization": "Bearer " + _LLM_chat_api_key}
17+
headers = {"Authorization": "Bearer " + _LLM_chat_api_key}
1818

1919

2020
def get_embedder():
21-
return OllamaEmbeddings(base_url=_LLM_BASE, model=_LLM_model, headers={'headers': headers})
21+
return OllamaEmbeddings(base_url=_LLM_BASE, model=_LLM_model, headers=headers)
2222

2323
def get_generator():
24-
return Ollama(base_url=_LLM_chat_base, model=_LLM_chat_model, temperature=0.0, headers={'headers': headers})
24+
return Ollama(base_url=_LLM_chat_base, model=_LLM_chat_model, temperature=0.0, headers=headers)
2525

2626
def get_chat():
2727
return ChatOllama(base_url=_LLM_chat_base, model=_LLM_chat_model, temperature=0.0, client_kwargs={'headers': headers})

0 commit comments

Comments
 (0)