Merge pull request #30 from andri-jpg/main

use pydantic.v1 in langchain.py

Author: LLukas22

llm_rs/langchain/langchain.py

@@ -9,7 +9,10 @@
 from typing import Any, Dict, Optional, Sequence, Union, List
 import os
 
-from pydantic import root_validator
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
 
 from ..auto import AutoModel, KnownModels
 from ..config import GenerationConfig, SessionConfig
@@ -121,4 +124,4 @@ def embed_documents(self, texts: List[str]) -> List[List[float]]:
 
     def embed_query(self, text: str) -> List[float]:
         """Embed query text."""
-        return self.model.embed(text)
+        return self.model.embed(text)

pyproject.toml

@@ -18,7 +18,7 @@ documentation = "https://llukas22.github.io/llm-rs-python/"
 
 [project.optional-dependencies]
 convert = [ "transformers >= 4.29.0", "sentencepiece >= 0.1.99", "torch >= 2.0.0", "accelerate >= 0.19.0", "tqdm", "einops >= 0.6.1",]
-langchain = [ "langchain>=0.0.190",]
+langchain = [ "langchain>=0.0.266",]
 
 haystack=[
   "farm-haystack>=1.17.1"