dependency udpates and small adjustements handling misconfigured embedding or chatting options

Author: AndiMajore

nedrexapi/llm.py

@@ -3,18 +3,23 @@
 from nedrexapi.config import config as _config
 from langchain_community.llms.ollama import Ollama
 
-_LLM_BASE=_config["embeddings.server_base"]
-_LLM_model=_config[f"embeddings.model"]
-_LLM_path=_config[f"embeddings.path"]
+def get_conf_or_none(key):
+    try:
+        return _config.get(key)
+    except:
+        return None
 
-# _LLM_user=_config[f"embeddings.user"]
-# _LLM_pass=_config[f"embeddings.pass"]
+_LLM_BASE=get_conf_or_none("embeddings.server_base")
+_LLM_model=get_conf_or_none("embeddings.model")
+_LLM_path=get_conf_or_none("embeddings.path")
+
+_LLM_chat_model=get_conf_or_none("chat.model")
+_LLM_chat_base=get_conf_or_none("chat.server_base")
+_LLM_chat_api_key=get_conf_or_none("chat.api_key")
+
+headers = {"Authorization": "Bearer " + str(_LLM_chat_api_key)}
 
-_LLM_chat_model=_config["chat.model"]
-_LLM_chat_base=_config["chat.server_base"]
-_LLM_chat_api_key=_config["chat.api_key"]
 
-headers = {"Authorization": "Bearer " + _LLM_chat_api_key}
 
 
 def get_embedder():
@@ -24,15 +29,12 @@ def get_generator():
     return Ollama(base_url=_LLM_chat_base, model=_LLM_chat_model, temperature=0.0, headers=headers)
 
 def get_chat():
-    print(headers)
     return ChatOllama(base_url=_LLM_chat_base, model=_LLM_chat_model, temperature=0.0, client_kwargs={'headers': headers})
 
 def get_embedding(query):
     embedder = get_embedder()
     return embedder.embed(query)
 
-
-
 def generate(query):
     ollama_llm = get_generator()
     response = ollama_llm.invoke(query)

poetry.lock

@@ -41,7 +41,8 @@ langchain-neo4j = "^0.8.0"
 langchain-ollama = "^1.0.1"
 aiohttp = "^3.12.14"
 fonttools = "^4.60.2"
-pypdf = "^6.7.1"
+pypdf = "^6.8.0"
+langgraph = "^1.0.10"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.2.0"