chore: resolve langchain root dependency warning; move pytest to grou…

…p test dependency only
fynnfluegge · Jan 20, 2024 · 1f8a368 · 1f8a368
1 parent 773dcc7
commit 1f8a368
Show file tree

Hide file tree

Showing 3 changed files with 1,179 additions and 981 deletions.
diff --git a/doc_comments_ai/llm.py b/doc_comments_ai/llm.py
@@ -4,9 +4,10 @@
 from enum import Enum
 
 import inquirer
-from langchain import LLMChain, PromptTemplate
+from langchain.chains import LLMChain
 from langchain.chat_models import ChatLiteLLM
 from langchain.llms import LlamaCpp
+from langchain.prompts import PromptTemplate
 
 from doc_comments_ai import utils