blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rm pandas from arize (#6392)

Browse Source
This commit is contained in:
Harrison Chase 2023-06-18 18:33:04 -07:00 committed by GitHub
parent 3b7c4c51d5
commit 5ccebce777
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/callbacks/arize_callback.py
View File

@ -2,8 +2,6 @@
import uuid
from typing import Any, Dict, List, Optional, Union
import pandas as pd
from langchain.callbacks.base import BaseCallbackHandler
from langchain.schema import AgentAction, AgentFinish, LLMResult
@ -77,6 +75,7 @@ class ArizeCallbackHandler(BaseCallbackHandler):
def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
"""Log data to Arize when an LLM ends."""
import pandas as pd
from arize.utils.types import Embedding, Environments, ModelTypes
# Record token usage of the LLM