gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Larfii
2024-10-08 14:39:05 +08:00
parent 44463503fd
commit c18697305e
15 changed files with 174 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/batch_eval.py
View File

@@ -6,8 +6,8 @@ import jsonlines
from openai import OpenAI
def batch_eval(query_file, result1_file, result2_file, output_file_path, api_key):
client = OpenAI(api_key=api_key)
def batch_eval(query_file, result1_file, result2_file, output_file_path):
client = OpenAI()
with open(query_file, 'r') as f:
data = f.read()

2
examples/generate_query.py
View File

@@ -2,7 +2,7 @@ import os
from openai import OpenAI
os.environ["OPENAI_API_KEY"] = ""
# os.environ["OPENAI_API_KEY"] = ""
def openai_complete_if_cache(
model="gpt-4o-mini", prompt=None, system_prompt=None, history_messages=[], **kwargs

2
examples/insert.py
View File

@@ -3,7 +3,7 @@ import sys
from lightrag import LightRAG
os.environ["OPENAI_API_KEY"] = ""
# os.environ["OPENAI_API_KEY"] = ""
WORKING_DIR = ""

2
examples/query.py
View File

@@ -3,7 +3,7 @@ import sys
from lightrag import LightRAG, QueryParam
os.environ["OPENAI_API_KEY"] = ""
# os.environ["OPENAI_API_KEY"] = ""
WORKING_DIR = ""