From fe595320f93a4264dccfcd0306929b9467a4002d Mon Sep 17 00:00:00 2001 From: Cynthia Yang Date: Sun, 24 Sep 2023 20:02:30 -0700 Subject: [PATCH] Refactor (#5) --- .../integration_tests/llms/test_fireworks.py | 18 ++---------------- 1 file changed, 2 insertions(+), 16 deletions(-) diff --git a/libs/langchain/tests/integration_tests/llms/test_fireworks.py b/libs/langchain/tests/integration_tests/llms/test_fireworks.py index 3833430b5f77a..e450a90f6e351 100644 --- a/libs/langchain/tests/integration_tests/llms/test_fireworks.py +++ b/libs/langchain/tests/integration_tests/llms/test_fireworks.py @@ -1,22 +1,8 @@ """Test Fireworks AI API Wrapper.""" -<<<<<<< HEAD -from pathlib import Path - -import pytest - -from langchain.chains import RetrievalQA -from langchain.chains.llm import LLMChain -from langchain.document_loaders import TextLoader -from langchain.embeddings.openai import OpenAIEmbeddings -from langchain.llms import OpenAIChat -from langchain.llms.fireworks import Fireworks, FireworksChat -from langchain.llms.loading import load_llm -from langchain.prompts import PromptTemplate -======= from typing import Generator -from langchain import LLMChain, PromptTemplate +from langchain.chains import LLMChain +from langchain.prompts import PromptTemplate from langchain.llms.fireworks import Fireworks ->>>>>>> 1af2fefa9 (Refactor Fireworks and add ChatFireworks (#3)) from langchain.prompts.chat import ( ChatPromptTemplate, HumanMessagePromptTemplate,