diff --git a/agent/developer.py b/agent/developer.py
index 5b4b081..cbaa034 100644
--- a/agent/developer.py
+++ b/agent/developer.py
@@ -1,6 +1,6 @@
 from typing import Optional, Dict, Any
 from langchain_core.messages import HumanMessage, SystemMessage
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 import json
 from config import MODEL_NAME, TEMPERATURE, LLM_HOST
 from models import CodeSolution, ReviewResult, ReviewComment, Task
diff --git a/agent/reviewer.py b/agent/reviewer.py
index 9d9abda..0513465 100644
--- a/agent/reviewer.py
+++ b/agent/reviewer.py
@@ -1,6 +1,6 @@
 from typing import Optional, Dict, Any
 from langchain_core.messages import HumanMessage, SystemMessage
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 import json
 from config import MODEL_NAME, TEMPERATURE, LLM_HOST
 from models import CodeSolution, ReviewResult, ReviewComment, Task
diff --git a/requirements.txt b/requirements.txt
index e17e3b7..45eba5e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,5 +3,6 @@
 langgraph>=0.3.0
 langchain>=0.2.0
 langchain-community>=0.0.16
+langchain-ollama
 python-dotenv>=1.0.0
 pydantic>=2.0.0