mirror of
https://github.com/IntelligenzaArtificiale/Free-Auto-GPT.git
synced 2026-07-03 13:15:41 +02:00
IntelligenzaArtificiale
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
from langchain.llms.base import LLM
|
|
from typing import Optional, List, Mapping, Any
|
|
import t3nsor
|
|
|
|
class gpt3NoInternet(LLM):
|
|
messages: List[Mapping[str, Any]]
|
|
|
|
@property
|
|
def _llm_type(self) -> str:
|
|
return "custom"
|
|
|
|
def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
|
|
if stop is not None:
|
|
raise ValueError("stop kwargs are not permitted.")
|
|
|
|
t3nsor_cmpl = t3nsor.Completion.create(
|
|
prompt = prompt,
|
|
messages = self.messages
|
|
)
|
|
|
|
response = t3nsor_cmpl.completion.choices[0].text
|
|
|
|
self.messages.append({'role': 'user', 'content': prompt})
|
|
self.messages.append({'role': 'assistant', 'content': response})
|
|
|
|
return response
|
|
|
|
@property
|
|
def _identifying_params(self) -> Mapping[str, Any]:
|
|
"""Get the identifying parameters."""
|
|
return {"messages": self.messages}
|
|
|
|
|
|
#llm = gpt3NoInternet(messages=[])
|
|
|
|
#print(llm("Never forget you are a Python Programmer and I am a Stock Trader."))
|
|
|