Merge branch 'main' into dependabot/github_actions/actions/checkout-5

Author: souradipp76

Makefile

@@ -35,7 +35,7 @@ lint:             ## Run pep8, black, mypy linters.
 	$(ENV_PREFIX)flake8 readme_ready/
 	$(ENV_PREFIX)black -l 79 --check readme_ready/
 	$(ENV_PREFIX)black -l 79 --check tests/
-	$(ENV_PREFIX)mypy --ignore-missing-imports --disable-error-code=arg-type readme_ready/
+	$(ENV_PREFIX)mypy --ignore-missing-imports --disable-error-code=arg-type --disable-error-code=assignment readme_ready/
 
 .PHONY: test
 test: lint        ## Run tests and generate coverage report.

readme_ready/utils/llm_utils.py

@@ -129,6 +129,17 @@ def get_llama_chat_model(model_name: str, streaming=False, model_kwargs=None):
     )
 
 
+def get_ollama_chat_model(model_name: str, streaming=False, model_kwargs=None):
+    """Get Ollama Chat Model"""
+    from langchain_ollama import ChatOllama
+
+    return ChatOllama(
+        model=model_name,
+        temperature=model_kwargs["temperature"],
+        num_ctx=model_kwargs["max_length"],
+        disable_streaming=not streaming,
+    )
+
 def get_openai_chat_model(
     model: str, temperature=None, streaming=None, model_kwargs=None
 ) -> ChatOpenAI: