ChromaDB client Python 버전 이슈

1. ChromaDB python client는 아직 python 버전에 의존성이 있다.
2. python 3.10.12에서 테스트 해봤으며
3. python 3.11.5에서 작동한다는 git issue가 있다 링크
4. 빠르게 써보고 싶으면 google colab에서 테스트하길 추천

그외 chromdb 관련 이슈 링크

• 공식문서
• 개인블로그

ChromaDB 테스트 (Colab)

colab에 접속하고 한다. (https://colab.research.google.com/)

python 버전 확인 및 chromabDB설치

# python 버전 확인
! python3 --version
Python 3.10.12

# chromadb 설치
! pip install chromadb

chromadb 테스트

import chromadb

# chromadb client 생성
chroma_client = chromadb.Client()

# collection 생성
collection = chroma_client.create_collection(name="my_collection")

# collection 에 텍스트 추가
collection.add(
    documents=[
        "This is a document about pineapple",
        "This is a document about oranges"
    ],
    ids=["id1", "id2"]
)

# collection에 쿼리 날리기
results = collection.query(
    query_texts=["This is a query document about hawaii"], # Chroma will embed this for you
    n_results=2 # how many results to return
)
print(results)

# 또다른 쿼리 날려보기
results = collection.query(
    query_texts=["This is a query document about florida"], # Chroma will embed this for you
    n_results=2 # how many results to return
)
print(results)