Add extract_answer

languagemodels/__init__.py

@@ -1 +1 @@
-from .languagemodels import do, chat, is_positive, match, search
+from .languagemodels import do, chat, is_positive, match, search, extract_answer

languagemodels/languagemodels.py

@@ -133,6 +133,23 @@ def search(topic):
         return summary
 
 
+def extract_answer(question, context):
+    """Extract an answer to a question from a provided context
+
+    >>> extract_answer("What color is the ball?", "There is a green ball and a red box")
+    'green'
+    >>> extract_answer("Who created Python?", search('Python'))
+    'Guido van Rossum'
+    """
+
+    model = "distilbert-base-cased-distilled-squad"
+
+    if model not in modelcache:
+        modelcache[model] = pipeline("question-answering", model=model)
+
+    return modelcache[model](question, context)["answer"]
+
+
 def is_positive(doc):
     """Returns true of a supplied string is positive
 

readme.md

@@ -27,4 +27,7 @@ True
 
 >>> lm.search('Chemistry')
 'Chemistry is the scientific study...
+
+>>> lm.extract_answer("What color is the ball?", "There is a green ball and a red box")
+'green'
 ```