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
parent
f9fa32ca82
commit
111e371dad
1 changed files with 24 additions and 0 deletions
@ -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…
Reference in new issue