AlbertoFH98 commited on
Commit
887ecbd
·
1 Parent(s): 3d5c843

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -198,7 +198,7 @@ def get_gpt_response(transcription_path, query, logger):
198
  | StrOutputParser()
199
  )
200
  llm_output = rag_chain.invoke(query)
201
- dataset = llm.create_dataset(dataset_name="Sample LLM dataset", description="A dataset with LLM inputs and outputs", data_type="llm")
202
 
203
  client.create_example(
204
  inputs={"input": query},
@@ -214,7 +214,7 @@ def get_gpt_response(transcription_path, query, logger):
214
  dataset_name="Sample LLM dataset",
215
  llm_or_chain_factory=rag_chain,
216
  evaluation=evaluation_config,
217
- client=llm,
218
  )
219
  logger.info("Eval output!!!!")
220
  logger.info(eval_output)
 
198
  | StrOutputParser()
199
  )
200
  llm_output = rag_chain.invoke(query)
201
+ dataset = client.create_dataset(dataset_name="Sample LLM dataset", description="A dataset with LLM inputs and outputs", data_type="llm")
202
 
203
  client.create_example(
204
  inputs={"input": query},
 
214
  dataset_name="Sample LLM dataset",
215
  llm_or_chain_factory=rag_chain,
216
  evaluation=evaluation_config,
217
+ client=client,
218
  )
219
  logger.info("Eval output!!!!")
220
  logger.info(eval_output)