liumaolin commited on
Commit
9965918
·
1 Parent(s): 1abf080

Add dynamic library path setup for `TestLLMDialogue`

Browse files

- Insert `src` directory to `sys.path` dynamically for test imports.
- Update imports to include `Path` for path management.

Files changed (1) hide show
  1. tests/test_llm_dialogue.py +7 -0
tests/test_llm_dialogue.py CHANGED
@@ -1,4 +1,6 @@
 
1
  import unittest
 
2
 
3
  from langchain.chains.conversation.memory import ConversationBufferWindowMemory
4
  from langchain_community.chat_models.llamacpp import ChatLlamaCpp
@@ -7,6 +9,11 @@ from langchain_core.messages import SystemMessage
7
  from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, MessagesPlaceholder
8
  from langchain_core.runnables.history import RunnableWithMessageHistory
9
 
 
 
 
 
 
10
  from voice_dialogue.config import paths
11
  from voice_dialogue.config.llm_config import get_llm_model_params
12
 
 
1
+ import sys
2
  import unittest
3
+ from pathlib import Path
4
 
5
  from langchain.chains.conversation.memory import ConversationBufferWindowMemory
6
  from langchain_community.chat_models.llamacpp import ChatLlamaCpp
 
9
  from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, MessagesPlaceholder
10
  from langchain_core.runnables.history import RunnableWithMessageHistory
11
 
12
+ HERE = Path(__file__).parent.parent
13
+ lib_path = HERE / "src"
14
+ if lib_path.exists() and lib_path.as_posix() not in sys.path:
15
+ sys.path.insert(0, lib_path.as_posix())
16
+
17
  from voice_dialogue.config import paths
18
  from voice_dialogue.config.llm_config import get_llm_model_params
19