Spaces:
Runtime error
Runtime error
File size: 1,943 Bytes
2d144e4 31a8d29 2d144e4 |
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 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import os
from langchain.llms import OpenAI
from langchain.prompts import PromptTemplate
from langchain.prompts.few_shot import FewShotPromptTemplate
from langchain.chains import LLMChain
from langchain.chains import SimpleSequentialChain
from prompts import prompt_transcribe, prompt_command, examples_few_shot
class FewShot4UAVs:
def __init__(self) -> None:
self.llm = OpenAI(model_name="text-davinci-003", temperature=0.0)
def get_transcription(self, text):
transcribe_prompt = PromptTemplate(
input_variables=["text"],
template=prompt_transcribe
)
sentence_chain = LLMChain(llm=self.llm,
prompt=transcribe_prompt,
output_key="sentence")
return sentence_chain
def format_command(self, chain):
command = PromptTemplate(
input_variables=["sentence", "command"],
template="sentence: {sentence}\n{command}"
)
few_shot_prompt = FewShotPromptTemplate(
examples=examples_few_shot,
example_prompt=command,
suffix="sentence: {sentence}",
input_variables=["sentence"]
)
command_chain = LLMChain(llm=self.llm, prompt=few_shot_prompt, output_key="output")
sentence_chain = chain
sentence_command_chain = SimpleSequentialChain(
chains=[sentence_chain, command_chain], verbose=True
)
return sentence_command_chain
def to_file(self, text):
try:
with open("commands.txt", "w") as f:
f.write(text)
print("Command successfully written to the file.")
except Exception as e:
print(f"Error: {e}")
def get_command(self, text):
command = self.format_command(self.get_transcription(text))
self.to_file(command.run(text))
return command.run(text)
|