viho-llm 0.1.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 qiaowenbin<uikoo9@qq.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,100 @@
+# viho-llm
+
+Utility library for working with Google Gemini AI, providing common tools and helpers for AI interactions.
+
+## Installation
+
+```bash
+npm install viho-llm
+```
+
+## Prerequisites
+
+You need to have a Google AI API key. Get one from [Google AI Studio](https://makersuite.google.com/app/apikey).
+
+## Usage
+
+### Basic Example
+
+```javascript
+import { Gemini } from 'viho-llm';
+
+// Initialize Gemini client
+const gemini = Gemini({
+  apiKey: 'your-google-api-key',
+  modelName: 'gemini-pro',
+});
+
+// Send a chat message
+const response = await gemini.chat({
+  contents: [
+    {
+      role: 'user',
+      parts: [{ text: 'Hello, how are you?' }],
+    },
+  ],
+});
+
+console.log(response);
+```
+
+## API Reference
+
+### `Gemini(options)`
+
+Creates a new Gemini client instance.
+
+#### Parameters
+
+- `options` (Object) - Configuration options
+  - `apiKey` (string) **required** - Your Google AI API key
+  - `modelName` (string) **required** - Model name (e.g., 'gemini-pro', 'gemini-pro-vision')
+
+#### Returns
+
+Returns a Gemini client object with the following methods:
+
+##### `client.chat(chatOptions)`
+
+Sends a chat request to the Gemini API.
+
+**Parameters:**
+
+- `chatOptions` (Object)
+  - `contents` (Array) **required** - Array of message objects
+    - `role` (string) - Either 'user' or 'model'
+    - `parts` (Array) - Array of content parts
+      - `text` (string) - The text content
+
+**Returns:**
+
+- (Promise\<string\>) - The generated text response
+
+**Example:**
+
+```javascript
+const response = await gemini.chat({
+  contents: [
+    {
+      role: 'user',
+      parts: [{ text: 'Write a haiku about coding' }],
+    },
+  ],
+});
+```
+
+## License
+
+MIT
+
+## Author
+
+uikoo9 <uikoo9@qq.com>
+
+## Related Projects
+
+- [viho](https://github.com/uikoo9/viho) - Main CLI tool for managing AI models
+
+## Contributing
+
+Issues and pull requests are welcome on [GitHub](https://github.com/uikoo9/viho/issues).

package/index.js

@@ -0,0 +1,77 @@
+'use strict';
+
+var genai = require('@google/genai');
+var qiao_log_js = require('qiao.log.js');
+
+// gemini
+const logger = qiao_log_js.Logger('viho-llm');
+
+/**
+ * Gemini
+ * @param {*} options
+ * @returns
+ */
+const Gemini = (options) => {
+  const methodName = 'Gemini';
+
+  // check
+  if (!options) {
+    logger.info(methodName, 'need options');
+    return;
+  }
+  if (!options.apiKey) {
+    logger.info(methodName, 'need options.apiKey');
+    return;
+  }
+  if (!options.modelName) {
+    logger.info(methodName, 'need options.modelName');
+    return;
+  }
+
+  // gemini
+  const gemini = {};
+  gemini.client = new genai.GoogleGenAI({
+    vertexai: true,
+    apiKey: options.apiKey,
+  });
+
+  // chat
+  gemini.chat = async (chatOptions) => {
+    return await chat(gemini.client, options.modelName, chatOptions);
+  };
+
+  // r
+  return gemini;
+};
+
+// chat
+async function chat(client, modelName, chatOptions) {
+  const methodName = 'Gemini - chat';
+
+  // check
+  if (!chatOptions) {
+    logger.info(methodName, 'need chatOptions');
+    return;
+  }
+  if (!chatOptions.contents) {
+    logger.info(methodName, 'need chatOptions.contents');
+    return;
+  }
+
+  try {
+    const response = await client.models.generateContent({
+      model: modelName,
+      contents: chatOptions.contents,
+    });
+    if (!response || !response.text) {
+      logger.error(methodName, 'invalid response');
+      return;
+    }
+
+    return response.text;
+  } catch (error) {
+    logger.error(methodName, 'error', error);
+  }
+}
+
+exports.Gemini = Gemini;

package/package.json

@@ -0,0 +1,64 @@
+{
+  "name": "viho-llm",
+  "version": "0.1.1",
+  "description": "Utility library for working with Google Gemini AI, providing common tools and helpers for AI interactions",
+  "keywords": [
+    "llm",
+    "ai",
+    "gemini",
+    "google-ai",
+    "google-gemini",
+    "genai",
+    "ai-tools",
+    "language-model",
+    "ai-utilities",
+    "viho"
+  ],
+  "author": "uikoo9 <uikoo9@qq.com>",
+  "license": "MIT",
+  "homepage": "https://github.com/uikoo9/viho#readme",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/uikoo9/viho.git"
+  },
+  "bugs": {
+    "url": "https://github.com/uikoo9/viho/issues"
+  },
+  "main": "index.js",
+  "module": "src/index.js",
+  "sideEffects": false,
+  "files": [
+    "src",
+    "index.js",
+    "LICENSE",
+    "README.md"
+  ],
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "scripts": {
+    "build": "qpro rollup ./rollup.config.js"
+  },
+  "dependencies": {
+    "@google/genai": "^1.34.0",
+    "qiao.log.js": "^3.7.5"
+  },
+  "nx": {
+    "namedInputs": {
+      "default": [
+        "{projectRoot}/src/**/*"
+      ]
+    },
+    "targets": {
+      "build": {
+        "inputs": [
+          "default"
+        ],
+        "outputs": [
+          "{projectRoot}/index.js"
+        ]
+      }
+    }
+  },
+  "gitHead": "30ba092c8b2ab0e52894506eb93fdd77a26e4e36"
+}

package/src/gemini.js

@@ -0,0 +1,74 @@
+// gemini
+import { GoogleGenAI } from '@google/genai';
+
+// Logger
+import { Logger } from 'qiao.log.js';
+const logger = Logger('viho-llm');
+
+/**
+ * Gemini
+ * @param {*} options
+ * @returns
+ */
+export const Gemini = (options) => {
+  const methodName = 'Gemini';
+
+  // check
+  if (!options) {
+    logger.info(methodName, 'need options');
+    return;
+  }
+  if (!options.apiKey) {
+    logger.info(methodName, 'need options.apiKey');
+    return;
+  }
+  if (!options.modelName) {
+    logger.info(methodName, 'need options.modelName');
+    return;
+  }
+
+  // gemini
+  const gemini = {};
+  gemini.client = new GoogleGenAI({
+    vertexai: true,
+    apiKey: options.apiKey,
+  });
+
+  // chat
+  gemini.chat = async (chatOptions) => {
+    return await chat(gemini.client, options.modelName, chatOptions);
+  };
+
+  // r
+  return gemini;
+};
+
+// chat
+async function chat(client, modelName, chatOptions) {
+  const methodName = 'Gemini - chat';
+
+  // check
+  if (!chatOptions) {
+    logger.info(methodName, 'need chatOptions');
+    return;
+  }
+  if (!chatOptions.contents) {
+    logger.info(methodName, 'need chatOptions.contents');
+    return;
+  }
+
+  try {
+    const response = await client.models.generateContent({
+      model: modelName,
+      contents: chatOptions.contents,
+    });
+    if (!response || !response.text) {
+      logger.error(methodName, 'invalid response');
+      return;
+    }
+
+    return response.text;
+  } catch (error) {
+    logger.error(methodName, 'error', error);
+  }
+}

package/src/index.js

@@ -0,0 +1 @@
+export * from './gemini.js';