This commit is contained in:
Nuno Campos 2023-08-24 16:54:20 +02:00
parent 5426712311
commit 9a07032055

View File

@ -1,7 +1,6 @@
from operator import itemgetter from operator import itemgetter
from typing import Any, Dict, List, Optional, Union from typing import Any, Dict, List, Optional, Union
from uuid import UUID from uuid import UUID
from langchain.callbacks.tracers.stdout import ConsoleCallbackHandler
import pytest import pytest
from freezegun import freeze_time from freezegun import freeze_time
@ -12,6 +11,7 @@ from langchain import PromptTemplate
from langchain.callbacks.manager import Callbacks from langchain.callbacks.manager import Callbacks
from langchain.callbacks.tracers.base import BaseTracer from langchain.callbacks.tracers.base import BaseTracer
from langchain.callbacks.tracers.schemas import Run from langchain.callbacks.tracers.schemas import Run
from langchain.callbacks.tracers.stdout import ConsoleCallbackHandler
from langchain.chat_models.fake import FakeListChatModel from langchain.chat_models.fake import FakeListChatModel
from langchain.llms.fake import FakeListLLM, FakeStreamingListLLM from langchain.llms.fake import FakeListLLM, FakeStreamingListLLM
from langchain.load.dump import dumpd, dumps from langchain.load.dump import dumpd, dumps