Skip to content

Commit 84472ff

Browse files
committed
review comment
Signed-off-by: Akihiko Kuroda <akihikokuroda2020@gmail.com>
1 parent c18c9b2 commit 84472ff

1 file changed

Lines changed: 2 additions & 7 deletions

File tree

  • mellea/stdlib/requirements

mellea/stdlib/requirements/rag.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from ...backends.adapters import AdapterMixin
77
from ...core import Backend, Context, Requirement, ValidationResult
88
from ..components import Document, Message
9+
from ..context import ChatContext
910

1011

1112
class CitationMode(Enum):
@@ -191,17 +192,11 @@ async def validate(
191192
all_messages = ctx.as_list()
192193
if len(all_messages) > 1:
193194
# Rebuild context without last message
194-
# Import here to avoid circular dependency
195-
from ..context import ChatContext
196-
197195
context_before_response = ChatContext()
198196
for msg in all_messages[:-1]:
199197
context_before_response = context_before_response.add(msg)
200198
else:
201199
# If only one message, use empty context
202-
# Import here to avoid circular dependency
203-
from ..context import ChatContext
204-
205200
context_before_response = ChatContext()
206201

207202
# Handle empty response before calling intrinsic
@@ -215,7 +210,7 @@ async def validate(
215210

216211
# Call find_citations intrinsic
217212
try:
218-
# Import here to avoid circular dependency
213+
# Import here to avoid circular dependency with backends
219214
from ..components.intrinsic import rag
220215

221216
citations: list[dict] = rag.find_citations(

0 commit comments

Comments
 (0)