rm pandas from arize (#6392)

pull/6275/head
Harrison Chase 1 year ago committed by GitHub
parent 3b7c4c51d5
commit 5ccebce777
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save