Skip to content

Commit 058e187

Browse files
committed
fix: some correction in completions for robotframework >= 6.1
1 parent a7ba998 commit 058e187

File tree

1 file changed

+6
-2
lines changed
  • packages/language_server/src/robotcode/language_server/robotframework/parts

1 file changed

+6
-2
lines changed

packages/language_server/src/robotcode/language_server/robotframework/parts/completion.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -485,12 +485,16 @@ async def create_settings_completion_items(self, range: Optional[Range]) -> List
485485
from robot.parsing.lexer.settings import (
486486
InitFileSettings,
487487
ResourceFileSettings,
488-
TestCaseFileSettings,
489488
)
490489

490+
if get_robot_version() >= (6, 1):
491+
from robot.parsing.lexer.settings import SuiteFileSettings
492+
else:
493+
from robot.parsing.lexer.settings import TestCaseFileSettings as SuiteFileSettings
494+
491495
doc_type = await self.parent.documents_cache.get_document_type(self.document)
492496

493-
settings_class = TestCaseFileSettings
497+
settings_class = SuiteFileSettings
494498
if doc_type == DocumentType.RESOURCE:
495499
settings_class = ResourceFileSettings
496500
elif doc_type == DocumentType.INIT:

0 commit comments

Comments
 (0)