Merge pull request #51 from cp2boston/RCB-464_text-embedding

Rcb 464 text embedding

Author: hgaudet

examples/text_embedding.php

@@ -0,0 +1,27 @@
+<?php
+
+/**
+ * Example code to call Rosette API to get text vectors from sample text
+ **/
+require_once dirname(__FILE__) . '/vendor/autoload.php';
+use rosette\api\Api;
+use rosette\api\DocumentParameters;
+use rosette\api\RosetteException;
+
+$options = getopt(null, array('key:', 'url::'));
+if (!isset($options['key'])) {
+    echo 'Usage: php ' . __FILE__ . " --key <api_key> --url=<alternate_url>\n";
+    exit();
+}
+$embeddings_data = "Cambridge, Massachusetts";
+$api = isset($options['url']) ? new Api($options['key'], $options['url']) : new Api($options['key']);
+$params = new DocumentParameters();
+$content = $embeddings_data;
+$params->set('content', $content);
+
+try {
+    $result = $api->textEmbedding($params, false);
+    var_dump($result);
+} catch (RosetteException $e) {
+    error_log($e);
+}

source/rosette/api/Api.php

@@ -706,4 +706,18 @@ public function relationships($params)
     {
         return $this->callEndpoint($params, 'relationships');
     }
+
+    /**
+     * Calls the text-embedding endpoint.
+     *
+     * @param $params
+     *
+     * @return mixed
+     *
+     * @throws RosetteException
+     */
+    public function textEmbedding($params)
+    {
+        return $this->callEndpoint($params, 'text-embedding');
+    }
 }

spec/rosette/api/ApiSpec.php

@@ -203,6 +203,17 @@ public function it_calls_the_relationships_endpoint($params, $request)
         $this->setMockRequest($request);
         $this->relationships($params)->shouldHaveKeyWithValue('name', 'Rosette API');
     }
+    public function it_calls_the_text_embedding_endpoint($params, $request)
+    {
+        $params->beADoubleOf('\rosette\api\DocumentParameters');
+        $params->contentUri = 'http://some.dummysite.com';
+        $request->beADoubleOf('rosette\api\RosetteRequest');
+        $request->makeRequest(Argument::any(), Argument::any(), Argument::any(), Argument::any())->willReturn(true);
+        $request->getResponseCode()->willReturn(200);
+        $request->getResponse()->willReturn([ 'name' => 'Rosette API']);
+        $this->setMockRequest($request);
+        $this->textEmbedding($params)->shouldHaveKeyWithValue('name', 'Rosette API');
+    }
     public function it_fails_with_non_200_response($params, $request)
     {
         $params->beADoubleOf('\rosette\api\DocumentParameters');