Update langchain.py

toandaominh1997 · Dec 1, 2024 · 86c5a77 · 86c5a77
1 parent 2bc2309
commit 86c5a77
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/langchain.py b/langchain.py
@@ -1,13 +1,17 @@
 from typing import List, Union, Generator, Iterator
 import os 
-from langchain_core.prompts import ChatPromptTemplate
-from langchain_ollama.llms import OllamaLLM
+import pip
+
 
 class Pipeline:
     def __init__(self):
         self.index = 0
 
     async def on_startup(self):
+        pip.main(['install', "langchain"])
+        pip.main(['install', "langchain-ollama"])
+        from langchain_core.prompts import ChatPromptTemplate
+        from langchain_ollama.llms import OllamaLLM
         # This function is called when the server is started.
         template = """Question: {question}