Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions paperqa/llms.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
from __future__ import annotations

import asyncio
import itertools
import logging
Expand Down Expand Up @@ -32,13 +30,15 @@
Field,
model_validator,
)
from qdrant_client.http.models import Record
from typing_extensions import override

from paperqa.types import Doc, Text

if TYPE_CHECKING:
from qdrant_client.http.models import Record

from paperqa.docs import Docs

try:
from qdrant_client import AsyncQdrantClient, models

Expand Down Expand Up @@ -447,12 +447,12 @@ async def similarity_search(
@classmethod
async def load_docs(
cls,
client: AsyncQdrantClient,
client: "AsyncQdrantClient",
collection_name: str,
vector_name: str | None = None,
batch_size: int = 100,
max_concurrent_requests: int = 5,
) -> Docs:
) -> "Docs":
from paperqa.docs import Docs # Avoid circular imports

vectorstore = cls(
Expand Down
Loading