File size: 3,753 Bytes
0af0679 |
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 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
from dotenv import load_dotenv
from openai import OpenAI
import re, json
load_dotenv(override=True)
openai = OpenAI()
call_to_action = "Type something to manipulate, or 'exit' to quit."
def smart_capitalize(word):
for i, c in enumerate(word):
if c.isalpha():
return word[:i] + c.upper() + word[i+1:].lower()
return word # no letters to capitalize
def manipulate_string(input_string):
input_string = input_string[::-1]
words = re.split(r'\s+', input_string.strip())
capitalized_words = [smart_capitalize(word) for word in words]
return ' '.join(capitalized_words)
manipulate_string_json = {
"name": "manipulate_string",
"description": "Use this tool to reverse the characters in the text the user enters, then to capitalize the first letter of each reversed word)",
"parameters": {
"type": "object",
"properties": {
"input_string": {
"type": "string",
"description": "The text the user enters"
}
},
"required": ["input_string"],
"additionalProperties": False
}
}
tools = [{"type": "function", "function": manipulate_string_json}]
TOOL_FUNCTIONS = {
"manipulate_string": manipulate_string
}
def handle_tool_calls(tool_calls):
results = []
for tool_call in tool_calls:
tool_name = tool_call.function.name
arguments = json.loads(tool_call.function.arguments)
tool = TOOL_FUNCTIONS.get(tool_name)
result = tool(**arguments) if tool else {}
# Remove quotes if result is a plain string
content = result if isinstance(result, str) else json.dumps(result)
results.append({
"role": "tool",
"content": content,
"tool_call_id": tool_call.id
})
return results
system_prompt = f"""You are a helpful assistant who takes text from the user and manipulates it in various ways.
Currently you do the following:
- reverse the string the user entered
- convert to all lowercase letters so any words whose first letters were capitalized are now lowercase
- convert the first letter of each word in the reversed string to uppercase
Be professional, friendly and engaging, as if talking to a customer who came across your service.
Do not output any additional text, just the result of the string manipulation.
After outputting the text, prompt the user for the next input text with {call_to_action}
With this context, please chat with the user, always staying in character.
"""
def chat(message, history):
messages = [{"role": "system", "content": system_prompt}] + history + [{"role": "user", "content": message}]
done=False
while not done:
response = openai.chat.completions.create(model="gpt-4o-mini", messages=messages, tools=tools)
finish_reason = response.choices[0].finish_reason
if finish_reason == "tool_calls":
message = response.choices[0].message
tool_calls = message.tool_calls
results = handle_tool_calls(tool_calls)
messages.append(message)
messages.extend(results)
else:
done = True
return response.choices[0].message.content
def main():
print("\nWelcome to the string manipulation chat!")
print(f"{call_to_action}\n")
history = []
while True:
user_input = input("")
if user_input.lower() in {"exit", "quit"}:
print("\nThanks for using our service!")
break
response = chat(user_input, history)
history.append({"role": "user", "content": user_input})
history.append({"role": "assistant", "content": response})
print(response)
if __name__ == "__main__":
main()
|