support langchian v1

Author: JqRrt

cozeloop/integration/langchain/trace_callback.py

@@ -9,8 +9,9 @@
 
 import pydantic
 from pydantic import Field, BaseModel
-from langchain.callbacks.base import BaseCallbackHandler
-from langchain.schema import AgentFinish, AgentAction, LLMResult
+from langchain_core.callbacks.base import BaseCallbackHandler
+from langchain_core.outputs import LLMResult
+from langchain_core.agents import AgentFinish, AgentAction
 from langchain_core.prompt_values import PromptValue, ChatPromptValue
 from langchain_core.messages import BaseMessage, AIMessageChunk
 from langchain_core.prompts import AIMessagePromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate

cozeloop/integration/langchain/util.py

@@ -3,8 +3,7 @@
 
 import tiktoken
 from typing import List, Dict, Union, Any, Optional
-from langchain.schema import LLMResult
-from langchain_core.outputs import Generation, ChatGeneration
+from langchain_core.outputs import LLMResult, Generation, ChatGeneration
 
 
 def calc_token_usage(inputs: Union[List[Dict], LLMResult], model: str = 'gpt-3.5-turbo-0613'):