galaxybrain 0.0.1

LLM extensions framework

GalaxyBrain

Turn wordcel LLMs into mighty shape rotators!

Goals

1. Reduce surprises when it comes to working with LLMs.
2. Focus on production use cases and CI/CD compatibility.
3. Avoid bloat and keep base primitives simple.

Getting Started

First, install the library:

pip install galaxybrain

Currently, GalaxyBrain only supports OpenAI APIs, so you'll need to get an API key. You can then either keep it on your path as OPENAI_API_KEY or pass it to the completion object manually:

api_key = ...
OpenAiCompletion(api_key)

Here is an example of some of GalaxyBrain's functionality:

completion = OpenAiCompletion(api_key, temperature=0.9, user="demo")
memory = Memory()
rules = [
    json_rules.return_valid_json(),
    json_rules.put_answer_in_field("Names"),
    Rule("only use information from fantasy novels")
]

completion.complete(
    Prompt("Give me ideas for two names", memory=memory, rules=rules)
)

# Output:
# {
#     "Names": ["Frodo Baggins", "Gandalf the Grey"]
# }

result = completion.complete(
    Prompt("Give me 3 more from another universe", memory=memory)
)

# Output:
# {
#     "Names": ["Dumbledore", "Luna Lovegood", "Harry Potter"]
# }

You can also validate results against your rules:

validator = Validator(result, rules)

if validator.validate():
    log.info("Rule validations passed")
else:
    log.error("Rule validations failed")
    log.error(validator.failed_rules())