import jsonlines
import json
# import docx
# from deepeval.models import OllamaModel
import ollama
import time
#ollama run Hudson/llemma:7b
#deepeval set-ollama Hudson/llemma:7b

if __name__=="__main__":
    # Initialize parser

    tic = time.perf_counter()
    ollama.generate(model="Hudson/llemma:7b", prompt="I need to determine the radius of convergence of the series $\\sum_{n=1}^\\infty a_nx^n$, where $a_n=a^n+b^n$ and $a,b$ are real numbers.")
    # print(response)
    toc = time.perf_counter()
    print(f"Ran Llemma in {toc - tic:0.4f} seconds")