klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 13:25:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix typing of generated_fiels parameters

Browse Source
This commit is contained in:
sabaimran
2024-11-28 20:15:10 -08:00
parent d91935c880
commit 6f408948d3
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/khoj/processor/conversation/anthropic/anthropic_chat.py
View File

@@ -24,7 +24,7 @@ from khoj.utils.helpers import (
is_none_or_empty, is_none_or_empty,
truncate_code_context, truncate_code_context,
) )
from khoj.utils.rawconfig import LocationData from khoj.utils.rawconfig import FileAttachment, LocationData
from khoj.utils.yaml import yaml_dump from khoj.utils.yaml import yaml_dump
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -159,7 +159,7 @@ def converse_anthropic(
vision_available: bool = False, vision_available: bool = False,
query_files: str = None, query_files: str = None,
generated_images: Optional[list[str]] = None, generated_images: Optional[list[str]] = None,
generated_files: List[str] = None, generated_files: List[FileAttachment] = None,
generated_excalidraw_diagram: Optional[str] = None, generated_excalidraw_diagram: Optional[str] = None,
additional_context: Optional[str] = None, additional_context: Optional[str] = None,
tracer: dict = {}, tracer: dict = {},

4
src/khoj/processor/conversation/google/gemini_chat.py
View File

@@ -24,7 +24,7 @@ from khoj.utils.helpers import (
is_none_or_empty, is_none_or_empty,
truncate_code_context, truncate_code_context,
) )
from khoj.utils.rawconfig import LocationData from khoj.utils.rawconfig import FileAttachment, LocationData
from khoj.utils.yaml import yaml_dump from khoj.utils.yaml import yaml_dump
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -169,7 +169,7 @@ def converse_gemini(
vision_available: bool = False, vision_available: bool = False,
query_files: str = None, query_files: str = None,
generated_images: Optional[list[str]] = None, generated_images: Optional[list[str]] = None,
generated_files: List[str] = None, generated_files: List[FileAttachment] = None,
generated_excalidraw_diagram: Optional[str] = None, generated_excalidraw_diagram: Optional[str] = None,
additional_context: List[str] = None, additional_context: List[str] = None,
tracer={}, tracer={},

4
src/khoj/processor/conversation/offline/chat_model.py
View File

@@ -26,7 +26,7 @@ from khoj.utils.helpers import (
is_promptrace_enabled, is_promptrace_enabled,
truncate_code_context, truncate_code_context,
) )
from khoj.utils.rawconfig import LocationData from khoj.utils.rawconfig import FileAttachment, LocationData
from khoj.utils.yaml import yaml_dump from khoj.utils.yaml import yaml_dump
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -162,7 +162,7 @@ def converse_offline(
user_name: str = None, user_name: str = None,
agent: Agent = None, agent: Agent = None,
query_files: str = None, query_files: str = None,
generated_files: List[str] = None, generated_files: List[FileAttachment] = None,
additional_context: List[str] = None, additional_context: List[str] = None,
tracer: dict = {}, tracer: dict = {},
) -> Union[ThreadedGenerator, Iterator[str]]: ) -> Union[ThreadedGenerator, Iterator[str]]:

4
src/khoj/processor/conversation/openai/gpt.py
View File

@@ -22,7 +22,7 @@ from khoj.utils.helpers import (
is_none_or_empty, is_none_or_empty,
truncate_code_context, truncate_code_context,
) )
from khoj.utils.rawconfig import LocationData from khoj.utils.rawconfig import FileAttachment, LocationData
from khoj.utils.yaml import yaml_dump from khoj.utils.yaml import yaml_dump
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -158,7 +158,7 @@ def converse(
vision_available: bool = False, vision_available: bool = False,
query_files: str = None, query_files: str = None,
generated_images: Optional[list[str]] = None, generated_images: Optional[list[str]] = None,
generated_files: List[str] = None, generated_files: List[FileAttachment] = None,
generated_excalidraw_diagram: Optional[str] = None, generated_excalidraw_diagram: Optional[str] = None,
additional_context: List[str] = None, additional_context: List[str] = None,
tracer: dict = {}, tracer: dict = {},