pipelines/pipeline.py
Timothy J. Baek f91d48aff8 refac
2024-05-21 14:53:35 -07:00

14 lines
341 B
Python

from typing import List, Union, Generator
from schemas import OpenAIChatMessage
def get_response(
user_message: str, messages: List[OpenAIChatMessage]
) -> Union[str, Generator]:
# This is where you can add your custom pipelines like RAG.
print(messages)
print(user_message)
return f"rag response to: {user_message}"