Add openai tools

fmtr/tools/__init__.py

@@ -127,6 +127,11 @@
 except ImportError as exception:
     interface = MissingExtraMockModule('interface', exception)
 
+try:
+    from fmtr.tools import openai_tools as openai
+except ImportError as exception:
+    openai = MissingExtraMockModule('openai', exception)
+
 
 __all__ = [
     'config',

fmtr/tools/openai_tools.py

@@ -0,0 +1,40 @@
+from functools import lru_cache
+from openai import OpenAI
+
+from fmtr.tools import environment_tools
+
+OPENAI_TOKEN_KEY = 'FMTR_OPENAI_API_KEY'
+
+
+@lru_cache
+def get_client():
+    """
+
+    Get an OpenAI client instance
+
+    """
+    OPENAI_KEY = environment_tools.get(OPENAI_TOKEN_KEY)
+    client = OpenAI(api_key=OPENAI_KEY)
+    return client
+
+
+def get_text(prompt, model='gpt-4o'):
+    """
+
+    Very simple prompt-to-output
+
+    """
+    client = get_client()
+    messages = [{"role": 'user', "content": prompt}]
+
+    completion = client.chat.completions.create(
+        messages=messages,
+        model=model
+    )
+    text = completion.choices[0].message.content
+    return text
+
+
+if __name__ == '__main__':
+    text = get_text('hi')
+    text

fmtr/tools/version

@@ -1 +1 @@
-0.9.7
+0.9.8

requirements.py

@@ -28,6 +28,7 @@
     'metric': ['pandas'],
     'html': ['html2text'],
     'interface': ['streamlit'],
+    'openai.api': ['openai']
 }
 
 CONSOLE_SCRIPTS = [