SimpleRAGPipeline.predict:v0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import weave
@weave.op()
def predict(self, query: str):
"""
Predicts a response based on the input query.
Args:
query (str): The input query string.
Returns:
The generated response based on the retrieved context.
"""
context = self.retriever.predict(query, self.top_k)
return self.response_generator.predict(query, context)