向量库改为FAISS，解决Chroma和Google库里对pb不同版本的依赖问题

agent.py

@@ -9,9 +9,13 @@ from langchain_core.messages import HumanMessage, SystemMessage
 from langchain_ollama import ChatOllama
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_huggingface import HuggingFaceEmbeddings, ChatHuggingFace, HuggingFaceEndpoint
-from langchain_chroma import Chroma
 from langgraph.graph import START, StateGraph, MessagesState
 
+# from langchain_chroma import Chroma
+import faiss
+from langchain_community.docstore.in_memory import InMemoryDocstore
+from langchain_community.vectorstores import FAISS
+
 from langgraph.prebuilt import ToolNode
 from langgraph.prebuilt import tools_condition
 
@@ -134,10 +138,21 @@ sys_msg = SystemMessage(content=system_prompt)
 
 # Retriever
 embeddings = HuggingFaceEmbeddings(model_name="BAAI/bge-small-en-v1.5")
-vector_store = Chroma(
-    collection_name="demo_collection",
+
+# vector_store = Chroma(
+#     collection_name="demo_collection",
+#     embedding_function=embeddings,
+#     persist_directory="./chroma_langchain_db",
+# )
+
+embedding_dim = len(embeddings.embed_query("hello world"))
+index = faiss.IndexFlatL2(embedding_dim)
+
+vector_store = FAISS(
     embedding_function=embeddings,
-    persist_directory="./chroma_langchain_db",
+    index=index,
+    docstore=InMemoryDocstore(),
+    index_to_docstore_id={},
 )
 create_retriever_tool = create_retriever_tool(
     retriever= vector_store.as_retriever(),

requirements.txt

@@ -1,164 +1,13 @@
-aiofiles==24.1.0
-aiohappyeyeballs==2.6.1
-aiohttp==3.12.12
-aiosignal==1.3.2
-annotated-types==0.7.0
-anyio==4.9.0
-asgiref==3.8.1
-attrs==25.3.0
-backoff==2.2.1
-bcrypt==4.3.0
-beautifulsoup4==4.13.4
-bs4==0.0.2
-build==1.2.2.post1
-cachetools==5.5.2
-certifi==2025.4.26
-charset-normalizer==3.4.2
-chromadb==1.0.12
-click==8.2.1
-coloredlogs==15.0.1
-dataclasses-json==0.6.7
-distro==1.9.0
-durationpy==0.10
-fastapi==0.115.9
-ffmpy==0.6.0
-filelock==3.18.0
-filetype==1.2.0
-flatbuffers==25.2.10
-frozenlist==1.7.0
-fsspec==2025.5.1
-google-ai-generativelanguage==0.6.18
-google-api-core==2.25.1
-google-auth==2.40.3
-googleapis-common-protos==1.70.0
-gradio==5.34.0
-gradio_client==1.10.3
-groovy==0.1.2
-grpcio==1.73.0
-grpcio-status==1.73.0
-h11==0.16.0
-hf-xet==1.1.3
-httpcore==1.0.9
-httptools==0.6.4
-httpx==0.28.1
-httpx-sse==0.4.0
-huggingface-hub==0.33.0
-humanfriendly==10.0
-idna==3.10
-importlib_metadata==8.7.0
-importlib_resources==6.5.2
-Jinja2==3.1.6
-joblib==1.5.1
-jsonpatch==1.33
-jsonpointer==3.0.0
-jsonschema==4.24.0
-jsonschema-specifications==2025.4.1
-kubernetes==33.1.0
-langchain==0.3.25
-langchain-chroma==0.2.4
-langchain-community==0.3.25
-langchain-core==0.3.65
-langchain-google-genai==2.1.5
-langchain-huggingface==0.3.0
-langchain-ollama==0.3.3
-langchain-text-splitters==0.3.8
-langgraph==0.4.8
-langgraph-checkpoint==2.0.26
-langgraph-prebuilt==0.2.2
-langgraph-sdk==0.1.70
-langsmith==0.3.45
-markdown-it-py==3.0.0
-MarkupSafe==3.0.2
-marshmallow==3.26.1
-mdurl==0.1.2
-mmh3==5.1.0
-mpmath==1.3.0
-multidict==6.4.4
-mypy_extensions==1.1.0
-networkx==3.5
-numpy==2.3.0
-oauthlib==3.2.2
-ollama==0.5.1
-onnxruntime==1.22.0
-opentelemetry-api==1.34.1
-opentelemetry-exporter-otlp-proto-common==1.34.1
-opentelemetry-exporter-otlp-proto-grpc==1.34.1
-opentelemetry-instrumentation==0.55b1
-opentelemetry-instrumentation-asgi==0.55b1
-opentelemetry-instrumentation-fastapi==0.55b1
-opentelemetry-proto==1.34.1
-opentelemetry-sdk==1.34.1
-opentelemetry-semantic-conventions==0.55b1
-opentelemetry-util-http==0.55b1
-orjson==3.10.18
-ormsgpack==1.10.0
-overrides==7.7.0
-packaging==24.2
-pandas==2.3.0
-pillow==11.2.1
-pip_search==0.0.13
-posthog==4.8.0
-propcache==0.3.2
-proto-plus==1.26.1
-protobuf==5.29.5
-pyasn1==0.6.1
-pyasn1_modules==0.4.2
-pydantic==2.11.5
-pydantic-settings==2.9.1
-pydantic_core==2.33.2
-pydub==0.25.1
-Pygments==2.19.1
-PyPika==0.48.9
-pyproject_hooks==1.2.0
-python-dateutil==2.9.0.post0
-python-dotenv==1.1.0
-python-multipart==0.0.20
-pytz==2025.2
-PyYAML==6.0.2
-referencing==0.36.2
-regex==2024.11.6
-requests==2.32.4
-requests-oauthlib==2.0.0
-requests-toolbelt==1.0.0
-rich==14.0.0
-rpds-py==0.25.1
-rsa==4.9.1
-ruff==0.11.13
-safehttpx==0.1.6
-safetensors==0.5.3
-scikit-learn==1.7.0
-scipy==1.15.3
-semantic-version==2.10.0
-sentence-transformers==4.1.0
-setuptools==80.9.0
-shellingham==1.5.4
-six==1.17.0
-sniffio==1.3.1
-soupsieve==2.7
-SQLAlchemy==2.0.41
-starlette==0.45.3
-sympy==1.14.0
-tenacity==9.1.2
-threadpoolctl==3.6.0
-tokenizers==0.21.1
-tomlkit==0.13.3
-torch==2.7.1
-tqdm==4.67.1
-transformers==4.52.4
-typer==0.16.0
-typing-inspect==0.9.0
-typing-inspection==0.4.1
-typing_extensions==4.14.0
-tzdata==2025.2
-urllib3==2.4.0
-uvicorn==0.34.3
-uvloop==0.21.0
-watchfiles==1.0.5
-websocket-client==1.8.0
-websockets==15.0.1
-wikipedia==1.4.0
-wrapt==1.17.2
-xxhash==3.5.0
-yarl==1.20.1
-zipp==3.23.0
-zstandard==0.23.0
+gradio
+requests
+langchain
+langchain-community
+langchain-core
+langchain-google-genai
+langchain-huggingface
+sentence-transformers
+faiss-cpu
+langchain-ollama
+langgraph
+wikipedia
+python-dotenv

requirements.txt.backup

@@ -6,8 +6,7 @@ langchain-core
 langchain-google-genai
 langchain-huggingface
 sentence-transformers
-chromadb
-langchain-chroma
+faiss-cpu
 langchain-ollama
 langgraph
 wikipedia