Skip to content

Commit 6e704cf

Browse files
committed
fix: add v2 to agent
1 parent 2368871 commit 6e704cf

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

mindsdb_sdk/agents.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,9 +100,15 @@ def __init__(
100100
def completion(self, messages: List[dict]) -> AgentCompletion:
101101
return self.collection.completion(self.name, messages)
102102

103+
def completion_v2(self, messages: List[dict]) -> AgentCompletion:
104+
return self.collection.completion_v2(self.name, messages)
105+
103106
def completion_stream(self, messages: List[dict]) -> Iterable[object]:
104107
return self.collection.completion_stream(self.name, messages)
105108

109+
def completion_stream_v2(self, messages: List[dict]) -> Iterable[object]:
110+
return self.collection.completion_stream_v2(self.name, messages)
111+
106112
def add_files(self, file_paths: List[str], description: str, knowledge_base: str = None):
107113
"""
108114
Add a list of files to the agent for retrieval.

0 commit comments

Comments
 (0)