Spaces:
Runtime error
Runtime error
File size: 977 Bytes
1d63754 000213d 1d63754 e0cbbfe 1d63754 000213d 1d63754 e0cbbfe 1d63754 e0cbbfe 1d63754 e0cbbfe 1d63754 e0cbbfe 1d63754 e0cbbfe 000213d 1d63754 000213d e0cbbfe 1d63754 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
from langchain.base_language import BaseLanguageModel
from langchain.chat_models.openai import ChatOpenAI
from langchain.schema import AIMessage, OutputParserException
from codeinterpreterapi.prompts import remove_dl_link_prompt
async def remove_download_link(
input_response: str,
llm: BaseLanguageModel,
) -> str:
messages = remove_dl_link_prompt.format_prompt(input_response=input_response).to_messages()
message = await llm.apredict_messages(messages)
if not isinstance(message, AIMessage):
raise OutputParserException("Expected an AIMessage")
return message.content
async def test():
llm = ChatOpenAI(model="gpt-3.5-turbo-0613") # type: ignore
example = "I have created the plot to your dataset.\n\nLink to the file [here](sandbox:/plot.png)."
print(await remove_download_link(example, llm))
if __name__ == "__main__":
import asyncio
from dotenv import load_dotenv
load_dotenv()
asyncio.run(test())
|