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