Spaces:
Running
Running
File size: 1,220 Bytes
2319518 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
from abc import ABC, abstractmethod
from typing import Dict, Iterator, List, Optional, Union
from qwen_agent.llm.base import BaseChatModel
from qwen_agent.utils.utils import has_chinese_chars
# TODO: Should *planning* just be another action that uses other actions?
class Action(ABC):
def __init__(self, llm: BaseChatModel = None, stream: bool = False):
self.llm = llm
self.stream = stream
def run(self, *args, **kwargs) -> Union[str, Iterator[str]]:
if 'lang' not in kwargs:
if has_chinese_chars([args, kwargs]):
kwargs['lang'] = 'zh'
else:
kwargs['lang'] = 'en'
return self._run(*args, **kwargs)
@abstractmethod
def _run(self, *args, **kwargs) -> Union[str, Iterator[str]]:
raise NotImplementedError
# It is okay for an Action to not call LLMs.
def _call_llm(
self,
prompt: Optional[str] = None,
messages: Optional[List[Dict]] = None,
stop: Optional[List[str]] = None,
) -> Union[str, Iterator[str]]:
return self.llm.chat(
prompt=prompt,
messages=messages,
stop=stop,
stream=self.stream,
)
|