diff --git a/llm_sentiment.py b/llm_sentiment.py index ee8092b..8c990d3 100644 --- a/llm_sentiment.py +++ b/llm_sentiment.py @@ -1,3 +1,6 @@ +# This script goes through all speeches in the 'ev_speeches' collection and asks the AI if the speech is positive, negative or neutral towards electric vehicles. +# It also asks the AI to list all arguments related to electric vehicles in the speech and to give a detailed summary of what is said about electric vehicles in the speech. + from arango_class import ArangoDB from ollama_class import Ollama @@ -38,5 +41,5 @@ for speech in speeches: ''' summary = ollama.generate(prompt=prompt) speech['llm_summary'] = summary - + arango.update_ev_document(speech)