feat: Add script to fetch documents and generate arguments related to electric cars

This commit adds a new script, `example_fetch_docs.py`, which fetches documents from the ArangoDB database. The script uses the Ollama library to generate arguments related to electric cars from the fetched documents.

The script first queries the database to retrieve documents where the translation contains the phrase "electric car". It then iterates through the fetched documents and generates arguments using the Ollama library.

The generated arguments are based on a prompt that asks for arguments related to electric cars in the text of each document. The prompt is customized to provide a transcript of a speech given in the European Parliament.

This script will be useful for extracting arguments related to electric cars from a collection of documents.
main
Lasse Studion 2 years ago
parent f9fa32ca82
commit 111e371dad
  1. 24
      example_fetch_docs.py

@ -0,0 +1,24 @@
# Example of fetching documents and go though them one by one using Ollama:
from arango import ArangoClient
import ollama
from arango_ev_class import ArangoDB
# Get the documents where "electric car" is mentioned
arango = ArangoDB()
db = arango.db
q = 'FOR doc IN speeches FILTER doc.translation LIKE "%electric car%" RETURN doc'
cursor = db.aql.execute(q) # Modify query
documents = list(cursor)
# Go though the documents one by one
for doc in documents:
text = doc['translation']
prompt =f"""Below is a transcript of a speech given in the European Parliament. I'm interested in all arguments related to electric cars.
\n{text}\n
Please give me a list of all arguments related to electric cars in the text above. One argument per line.
Answer ONLY with the arguments. Kepp to the information in the text.""" #TODO Make a better prompt!
arguments = ollama.generate(prompt=prompt, model='llama3:8b-instruct-q5_K_M', options={'temperature': 0})
Loading…
Cancel
Save