Spaces:

akryldigital
/

audit_assistant

Sleeping

akryldigital commited on 22 days ago

Commit

7998a89

verified ·

1 Parent(s): e1240d1

fix templates

Files changed (1) hide show

src/llm/templates.py CHANGED Viewed

@@ -2,7 +2,11 @@
 from typing import List, Dict, Any, Union
 from dataclasses import dataclass
-from langchain.schema import SystemMessage, HumanMessage
 @dataclass

 from typing import List, Dict, Any, Union
 from dataclasses import dataclass
+try:
+    from langchain.schema import SystemMessage, HumanMessage
+except:
+    from langchain_core.messages import SystemMessage, HumanMessage
 @dataclass