import json
from ericsearch import EricSearch
from erictransformer import EricChat
eric_search = EricSearch()
with open("data.jsonl", "w", encoding="utf-8") as f:
sample_case = {"text": "This contains example data. It should contain at least two sentences."}
f.write(json.dumps(sample_case)+ "\n")
eric_search.train("data.jsonl")
eric_search = EricSearch(data_name="eric_search/")
eric_chat = EricChat(model_name="openai/gpt-oss-20b", eric_search=eric_search)
result = eric_chat("Tell me about artificial intelligence ")
print(result.text)