From b2942371dceb6f5c3bf5e222198fdaace3ab10dc Mon Sep 17 00:00:00 2001 From: Gormery Kombo Wanjiru Date: Mon, 5 Feb 2024 12:47:45 +0100 Subject: [PATCH] some rename --- requirements.txt | 3 ++- run.sh | 2 +- src/categorize_expenses.py | 4 ---- src/test.py | 8 ++++++++ 4 files changed, 11 insertions(+), 6 deletions(-) delete mode 100644 src/categorize_expenses.py create mode 100644 src/test.py diff --git a/requirements.txt b/requirements.txt index f67c8af..6bba818 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ fastapi uvicorn debugpy -langchain_community \ No newline at end of file +langchain_community +langchain \ No newline at end of file diff --git a/run.sh b/run.sh index fccefce..a17de61 100644 --- a/run.sh +++ b/run.sh @@ -1,5 +1,5 @@ python -m venv venv source venv/Scripts/activate pip install -r requirements.txt -python signal-processing.py +python src/test.py diff --git a/src/categorize_expenses.py b/src/categorize_expenses.py deleted file mode 100644 index a6809ed..0000000 --- a/src/categorize_expenses.py +++ /dev/null @@ -1,4 +0,0 @@ -from langchain_community.llms import Ollama - -llm = Ollama(model="llama2-uncensored") -print(llm.invoke("was neil armstrong gay")) diff --git a/src/test.py b/src/test.py new file mode 100644 index 0000000..ad8f20f --- /dev/null +++ b/src/test.py @@ -0,0 +1,8 @@ +from langchain.llms import Ollama +from langchain.callbacks.manager import CallbackManager +from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler + +llm = Ollama(model="llama2-uncensored", + callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])) + +llm("Tell me 5 facts about Roman history:") \ No newline at end of file